Create app.py

app.py

@@ -0,0 +1,161 @@
+import gradio as gr
+import google.generativeai as genai
+from github import Github
+import gitlab
+import requests
+import tempfile
+import docx
+
+def generate_guide(git_provider, repo_url, personal_access_token, gemini_api_key, guide_type):
+    try:
+        if git_provider == "GitHub":
+            g = Github(personal_access_token)
+            repo = g.get_repo(repo_url)
+            content = repo.get_contents("")
+            file_contents = ""
+            for file in content:
+                if file.name.endswith('.py'):
+                    file_contents += f"\n\n{file.name}:\n{file.decoded_content.decode()}"
+        elif git_provider == "GitLab":
+            gl = gitlab.Gitlab(url='https://gitlab.com', private_token=personal_access_token)
+            project = gl.projects.get(repo_url)
+            items = project.repository_tree(recursive=True)
+            file_contents = ""
+            for item in items:
+                if item['type'] == 'blob' and item['name'].endswith('.py'):
+                    file_content = project.files.get(item['path'], ref='main')
+                    file_contents += f"\n\n{item['name']}:\n{file_content.decode().decode()}"
+        elif git_provider == "Gitea":
+            base_url = "https://gitea.com/api/v1"
+            headers = {"Authorization": f"token {personal_access_token}"}
+            response = requests.get(f"{base_url}/repos/{repo_url}/contents", headers=headers)
+            response.raise_for_status()
+            file_contents = ""
+            for item in response.json():
+                if item['type'] == 'file' and item['name'].endswith('.py'):
+                    file_content = requests.get(item['download_url']).text
+                    file_contents += f"\n\n{item['name']}:\n{file_content}"
+        else:
+            return "Unsupported Git provider", None, None
+
+        genai.configure(api_key=gemini_api_key)
+        model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
+        
+        if guide_type == "User Guide":
+            prompt = f"""Based on the following Python code, generate a comprehensive user guide:
+
+            {file_contents}
+
+            Please organize the user guide into sections such as:
+            1. Introduction
+            2. Getting Started
+            3. Features and Functionality
+            4. How to Use
+            5. Tips and Best Practices
+            6. Troubleshooting
+            7. FAQ
+
+            For each section, provide step-by-step instructions on how to use the software's features.
+            Explain each functionality that the end user will interact with.
+            Include examples where appropriate.
+            
+            Important formatting instructions:
+            - The output should be in plain text
+            - Use clear section titles
+            - Number each step within sections
+            - Provide clear, concise instructions for each step
+            - Explain the purpose and benefit of each feature for non-technical users
+            - Do not include any installation or configuration instructions
+            """
+        else:  # Administration Guide
+            prompt = f"""Based on the following Python code, generate a comprehensive administration guide:
+
+            {file_contents}
+
+            Please organize the administration guide into sections such as:
+            1. System Overview
+            2. Security and Access Control
+            3. Performance Monitoring and Optimization
+            4. Backup and Recovery
+            5. Troubleshooting and Maintenance
+            6. Customization and Configuration
+            7. Best Practices for Administrators
+
+            For each section, provide detailed information and step-by-step instructions for system administrators.
+            Focus on backend operations, system management, and advanced configurations.
+            Include examples and potential scenarios an administrator might encounter.
+            
+            Important formatting instructions:
+            - The output should be in plain text
+            - Use clear section titles
+            - Number each step within sections
+            - Provide clear, concise instructions for each step
+            - Explain the purpose and implications of each administrative task
+            - Include any relevant security considerations
+            """
+
+        response = model.generate_content(prompt)
+        guide = response.text
+        
+        # Create DOCX file
+        doc = docx.Document()
+        doc.add_heading(guide_type, 0)
+        
+        for line in guide.split('\n'):
+            line = line.strip()
+            if line.endswith(':'):
+                doc.add_heading(line, level=1)
+            elif line.startswith('Step'):
+                doc.add_paragraph(line, style='List Number')
+            else:
+                doc.add_paragraph(line)
+        
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.docx') as temp_docx:
+            doc.save(temp_docx.name)
+            docx_path = temp_docx.name
+
+        # Create Markdown file
+        markdown_content = f"# {guide_type}\n\n"
+        for line in guide.split('\n'):
+            line = line.strip()
+            if line.endswith(':'):
+                markdown_content += f"\n## {line}\n\n"
+            elif line.startswith('Step'):
+                markdown_content += f"1. {line[5:]}\n"
+            else:
+                markdown_content += f"{line}\n"
+
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.md', mode='w', encoding='utf-8') as temp_md:
+            temp_md.write(markdown_content)
+            md_path = temp_md.name
+        
+        return guide, docx_path, md_path
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}", None, None
+
+iface = gr.Interface(
+    fn=generate_guide,
+    inputs=[
+        gr.Dropdown(
+            choices=["GitHub", "GitLab", "Gitea"],
+            label="Git Provider"
+        ),
+        gr.Textbox(label="Repository URL", placeholder="owner/repo"),
+        gr.Textbox(label="Personal Access Token", type="password"),
+        gr.Textbox(label="Gemini API Key", type="password"),
+        gr.Radio(["User Guide", "Administration Guide"], label="Guide Type")
+    ],
+    outputs=[
+        gr.Textbox(label="Generated Guide"),
+        gr.File(label="Download Guide (DOCX)"),
+        gr.File(label="Download Guide (Markdown)")
+    ],
+    title="Automated Guide Generator",
+    description="Generate a user guide or administration guide based on the Python code in a Git repository using Gemini AI. Select a Git provider, enter repository details, choose the guide type, and let AI create a comprehensive guide.",
+    allow_flagging="never",
+    theme="default",
+    analytics_enabled=False,
+)
+
+iface.launch()