Add json.loads() to handle result

app.py

@@ -1,3 +1,4 @@
+import json
 import requests
 import gradio as gr
 
@@ -11,7 +12,8 @@ def generate_readme(repo_url):
             timeout=120
         )
         if response.status_code == 200:
-            return response.json().get("result", "No result returned.")
+            data = json.loads(response.text)
+            return data["readme"]
         else:
             return f"Error: {response.status_code}\n{response.text}"
     except Exception as e:
@@ -19,14 +21,14 @@ def generate_readme(repo_url):
 
 DEFAULT_VALUE = (
     "## Hello, World!\n"
-    "Regardless of the outcome, it was **a truly informative and enjoyable hackathon.**   \n"
+    "Regardless of the result, it was **a truly informative and enjoyable hackathon.**   \n"
     "As someone who likes and is good at using Hugging Face, I hope there will be more opportunities like this! 🤗"
 )
 
 with gr.Blocks(gr.themes.Origin()) as demo:
     gr.HTML("<h1 style='text-align:center;'>🤖 ARA: Auto README.md Agent 📝</h1>")
     with gr.Tabs():
-        with gr.TabItem("README Generator"):
+        with gr.TabItem("README.md Generator"):
             repo_input = gr.Textbox(
                 label="GitHub Repository URL",
                 placeholder="Enter the GitHub repository URL (e.g. https://github.com/username/repo)"
@@ -60,68 +62,23 @@ with gr.Blocks(gr.themes.Origin()) as demo:
         with gr.TabItem("About"):
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("### Service Description")
-                    gr.Textbox(
-                        value="This service generates README.md files for GitHub repositories using AI agents.",
-                        lines=8,
-                        interactive=False
+                    gr.Markdown("# 👩‍🏫 Service Description")
+                    gr.Markdown(
+                        "This service generates README.md files for GitHub repositories using AI agents."
                     )
                 with gr.Column(scale=1):
-                    gr.Markdown("### Features")
-                    gr.Textbox(
-                        value="- Generate README.md from GitHub repo\n- Preview generated README\n- Demo video included",
-                        lines=4,
-                        interactive=False
+                    gr.Markdown("# 🔛 Current Features")
+                    gr.Markdown(
+                        "- Generate README.md from GitHub repo\n"
+                        "- Preview generated README\n"
+                        "- Demo video included"
                     )
-                    gr.Markdown("### Future Features")
-                    gr.Textbox(
-                        value="- Support for multiple branches\n- Enhanced code analysis\n- User authentication",
-                        lines=4,
-                        interactive=False
+                    gr.Markdown("# 🔜 Future Features")
+                    gr.Markdown(
+                        "- Support for multiple branches\n"
+                        "- Enhanced code analysis\n"
+                        "- User authentication"
                     )
-                        
-# with gr.Blocks() as demo:
-#     gr.HTML("<h1 style='text-align:center;'>🤖 ARA: Auto README.md Agent 📝</h1>")
-#     repo_input = gr.Textbox(
-#         label="GitHub Repository URL",
-#         placeholder="Enter the GitHub repository URL (e.g. https://github.com/username/repo)"
-#     )
-    
-#     generate_btn = gr.Button("Generate README.md")
-    
-#     DEFAULT_VALUE = (
-#         "## Hello, World!\n"
-#         "Regardless of the outcome, it was **a truly informative and enjoyable hackathon.**   \n"
-#         "As someone who likes and is good at using Hugging Face, I hope there will be more opportunities like this! 🤗"
-#     )
-    
-#     with gr.Row():
-#         with gr.Column():
-#             gr.Markdown("### 📝 Draft")
-#             readme_preview = gr.Textbox(
-#                 lines=24,
-#                 label="Generated by agent",
-#                 interactive=True,
-#                 value=DEFAULT_VALUE
-#             )
-#         with gr.Column():
-#             gr.Markdown("### 🖼️ Preview")
-#             readme_markdown = gr.Markdown(value=DEFAULT_VALUE, show_copy_button=True)
-
-#     generate_btn.click(
-#         generate_readme,
-#         inputs=repo_input,
-#         outputs=readme_preview
-#     )
-
-#     readme_preview.change(
-#         lambda md: md,
-#         inputs=readme_preview,
-#         outputs=readme_markdown
-#     )
 
 if __name__ == "__main__":
-    demo.launch()
-    
-    #Error: 422
-#{"detail":[{"type":"missing","loc":["body","repo_url"],"msg":"Field required","input":{"query":"https://github.com/nsbg/nsbg.github.io"}}]}
+    demo.launch()

server.py

@@ -1,17 +1,19 @@
 import modal
 
 HF_SECRET_NAME = "hf-secret"
-MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.2"
+MODEL_ID = "google/gemma-3-12b-it"
 
 image = (
     modal.Image.debian_slim(python_version="3.12")
+    .apt_install("git")
     .pip_install(
         "smolagents[toolkit]",
         "huggingface_hub",
         "transformers",
         "duckduckgo-search",
         "fastapi",
-        "uvicorn"
+        "uvicorn",
+        "GitPython"
     )
 )
 
@@ -23,51 +25,23 @@ app = modal.App("auto-readme-agent")
     secrets=[modal.Secret.from_name("hf-secret")],
     timeout=180,
 )
-# @modal.asgi_app()
-# def fastapi_app():
-#     from fastapi import FastAPI, HTTPException
-#     from pydantic import BaseModel
-#     from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
-
-#     class AgentRequest(BaseModel):
-#         query: str
-
-#     # Use DuckDuckGoSearchTool as an example tool
-#     agent = CodeAgent(
-#         tools=[DuckDuckGoSearchTool()],
-#         model=HfApiModel(
-#             model_id=MODEL_ID
-#         ),  # Uses Hugging Face Inference API with your token
-#         stream_outputs=True
-#     )
-
-#     app = FastAPI()
-
-#     @app.post("/")
-#     async def run_agent(req: AgentRequest):
-#         try:
-#             result = agent.run(req.query)
-#             return {"result": result}
-#         except Exception as e:
-#             raise HTTPException(status_code=500, detail=str(e))
-
-#     return app
 @modal.asgi_app()
 def fastapi_app():
-    from fastapi import FastAPI, HTTPException
-    from pydantic import BaseModel
-    from smolagents import CodeAgent, HfApiModel
     import os
     import tempfile
     import shutil
-    import subprocess
-    from git import Repo
 
+    from git import Repo
+    from fastapi import FastAPI, HTTPException
+    from pydantic import BaseModel
+    from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool
+    
     class RepoRequest(BaseModel):
         repo_url: str
 
     agent = CodeAgent(
         model=HfApiModel(),
+        tools=[],
         stream_outputs=True
     )
 
@@ -83,15 +57,24 @@ def fastapi_app():
             repo_summary.append(f"Directory: {rel_path}")
             
             for file in files:
-                if file.endswith(('.py', '.md', '.txt', '.json', '.yaml')):
+                if file.endswith(('.py')):
                     file_path = os.path.join(root, file)
                     try:
                         with open(file_path, 'r', encoding='utf-8') as f:
-                            content = f.read(1000)  # Read first 1000 characters
-                        repo_summary.append(f"  File: {file}\n    Content: {content[:500]}...")
-                    except Exception as e:
-                        repo_summary.append(f"  File: {file} [Error reading file]")
-        return "\n".join(repo_summary)
+                            content = f.read()  # Read first 1000 characters
+                            filtered_lines = [
+                                line for line in content.splitlines()
+                                if not (
+                                    line.strip().startswith('#')
+                                    or line.strip().startswith('import')
+                                    or line.strip().startswith('from')
+                                )
+                            ]
+                            filtered_content = "\n".join(filtered_lines)
+                        repo_summary.append(f"File: {file}\n    Content: {filtered_content}...")
+                    except Exception:
+                        raise f"File: {file} [Error reading file]"
+        return "\n".join(repo_summary)[:500]
 
     @app.post("/")
     async def generate_readme(req: RepoRequest):
@@ -102,26 +85,49 @@ def fastapi_app():
             
             # Analyze repository
             repo_analysis = analyze_repo(temp_dir)
-            
+
             # Create prompt
-            prompt = f"""Create a comprehensive README.md for this GitHub repository based on its structure and contents:
-            
-            Repository Structure:
-            {repo_analysis}
-            
-            The README should include:
-            - Project description
-            - Installation instructions
-            - Usage examples
-            - Contributing guidelines
-            - License information if available
-            
-            Format using markdown with proper sections."""
-            
+            prompt = f"""Create a comprehensive README.md for this GitHub repository based on its structure and contents.
+
+Repository Structure:
+{repo_analysis}
+
+It would be good if the README.md includes the following contents:
+- Repository name
+- Project description
+- Contributing guidelines
+
+Format using markdown with proper sections.
+
+Here is an example of the style and detail you should provide:
+---
+# ExampleProject
+
+A Python package for advanced data analysis and visualization.
+
+## Installation
+```
+pip install exampleproject
+```
+
+## Usage
+```
+from exampleproject import analyze
+
+analyze('data.csv')
+```
+
+## Contribution
+
+Contributions are welcome! Please open an issue or submit a pull request.
+
+...
+"""
+
             # Generate README
             result = agent.run(prompt)
-            return {"readme": result}
             
+            return {"readme": result}
         except Exception as e:
             raise HTTPException(status_code=500, detail=str(e))
         finally: