Upload folder using huggingface_hub

CriticalThinking/app/api/router.py

@@ -21,7 +21,7 @@ class AnalyzeRequest(BaseModel):
 class AnalyzeResponse(BaseModel):
     task_id: str
 
-async def run_analysis_task(task_id: str, repo_url: str, project_description: str):
+def run_analysis_task(task_id: str, repo_url: Optional[str], project_description: str):
     tasks[task_id]["status"] = "processing"
     try:
         api_key = os.getenv("OPENAI_API_KEY", "dummy")

CriticalThinking/app/services/indexer.py

@@ -28,13 +28,13 @@ class CodeIndexer:
         temp_dir = tempfile.mkdtemp()
         try:
             print(f"Cloning {repo_url} into {temp_dir}...")
-            if repo_url.startswith("local://"):
-                local_path = repo_url.replace("local://", "")
-                shutil.copytree(local_path, temp_dir, dirs_exist_ok=True)
-            else:
-                result = subprocess.run(["git", "clone", "--depth", "1", repo_url, temp_dir], capture_output=True, text=True)
-                if result.returncode != 0:
-                    raise Exception(f"Git clone failed: {result.stderr}")
+            # Only allow HTTP/HTTPS URLs for security
+            if not repo_url.startswith(("http://", "https://")):
+                raise Exception("Only HTTP and HTTPS repository URLs are allowed.")
+
+            result = subprocess.run(["git", "clone", "--depth", "1", repo_url, temp_dir], capture_output=True, text=True)
+            if result.returncode != 0:
+                raise Exception(f"Git clone failed: {result.stderr}")
 
             self._index_directory(temp_dir)
         finally:

CriticalThinking/tests/test_api.py

@@ -5,8 +5,13 @@ from unittest.mock import patch, MagicMock
 
 client = TestClient(app)
 
+@patch("app.api.router.CodeIndexer")
 @patch("app.api.router.WebResearcher")
-def test_analyze_flow(mock_web_researcher_class):
+def test_analyze_flow(mock_web_researcher_class, mock_indexer_class):
+    # Mock CodeIndexer
+    mock_indexer = MagicMock()
+    mock_indexer_class.return_value = mock_indexer
+    
     # Mock WebResearcher
     mock_web_researcher = MagicMock()
     mock_web_researcher_class.return_value = mock_web_researcher
@@ -14,7 +19,7 @@ def test_analyze_flow(mock_web_researcher_class):
     mock_web_researcher.research_hf_spaces.return_value = "Mocked HF Spaces results"
 
     # Submit analysis
-    response = client.post("/analyze", json={"repo_url": "local://.", "project_description": "Test Project"})
+    response = client.post("/analyze", json={"repo_url": "https://github.com/dummy/repo", "project_description": "Test Project"})
     assert response.status_code == 200
     task_id = response.json()["task_id"]
     assert task_id

hf_app.py

@@ -6,10 +6,40 @@ from CriticalThinking.app.main import app as fastapi_app
 # We can just mount it or use it as the main app.
 # Here we will mount Gradio onto the existing FastAPI app.
 
+import httpx
+
 def analyze_interface(repo_url, project_description):
-    # This is a placeholder for the Gradio UI to interact with the API
-    # In a real scenario, we might want to use the background task or just call the service.
-    return f"Analysis request for {repo_url} received. Please use the API endpoints to monitor progress."
+    if not repo_url and not project_description:
+        return "Please provide at least a project description."
+    
+    try:
+        # Call the local FastAPI endpoint
+        with httpx.Client() as client:
+            response = client.post(
+                "http://localhost:7860/analyze",
+                json={
+                    "repo_url": repo_url if repo_url else None,
+                    "project_description": project_description
+                },
+                timeout=10.0
+            )
+            response.raise_for_status()
+            data = response.json()
+            task_id = data.get("task_id")
+            return f"Analysis started! Task ID: {task_id}\n\nYou can check the report at: /report/{task_id}"
+    except Exception as e:
+        return f"Error starting analysis: {str(e)}"
+
+def get_report_interface(task_id):
+    if not task_id:
+        return "Please provide a Task ID."
+    try:
+        with httpx.Client() as client:
+            response = client.get(f"http://localhost:7860/report/{task_id}")
+            response.raise_for_status()
+            return response.json()
+    except Exception as e:
+        return f"Error fetching report: {str(e)}"
 
 with gr.Blocks(title="Critical Code Agent") as demo:
     gr.Markdown("# 🦀 Critical Code Agent")
@@ -23,6 +53,14 @@ with gr.Blocks(title="Critical Code Agent") as demo:
     output = gr.Textbox(label="Status")
     
     analyze_btn.click(analyze_interface, inputs=[repo_url, project_desc], outputs=output)
+
+    gr.Markdown("### Check Report Status")
+    with gr.Row():
+        task_id_input = gr.Textbox(label="Task ID")
+        report_btn = gr.Button("Get Report")
+    
+    report_output = gr.JSON(label="Analysis Report")
+    report_btn.click(get_report_interface, inputs=[task_id_input], outputs=report_output)
     
     gr.Markdown("### API Endpoints")
     gr.Markdown("- `POST /analyze`: Submit a repository for analysis")