Upload folder using huggingface_hub

CriticalThinking/app/api/router.py

@@ -15,7 +15,7 @@ router = APIRouter()
 tasks: Dict[str, Any] = {}
 
 class AnalyzeRequest(BaseModel):
-    repo_url: str
+    repo_url: Optional[str] = None
     project_description: Optional[str] = "Generic software project"
 
 class AnalyzeResponse(BaseModel):
@@ -32,8 +32,9 @@ async def run_analysis_task(task_id: str, repo_url: str, project_description: st
         matcher = HFMatcher()
         web_researcher = WebResearcher()
 
-        # 1. Index
-        indexer.index_repository(repo_url)
+        # 1. Index (if repo_url is provided)
+        if repo_url:
+            indexer.index_repository(repo_url)
 
         # 2. Analyze
         analysis_results = orchestrator.run_analysis(project_description)

CriticalThinking/app/services/agent_orchestrator.py

@@ -87,7 +87,11 @@ class AgentOrchestrator:
             sub_q = step.get("sub_question")
             # Search codebase
             results = self.indexer.search(sub_q, limit=3)
-            context = "\n---\n".join([r.get("text", "") for r in results])
+            if results:
+                context = "\n---\n".join([r.get("text", "") for r in results])
+            else:
+                # If no code results, use the sub-question itself as context for conceptual analysis
+                context = f"No code snippets found for: {sub_q}. Analyzing based on project overview: {project_overview}"
 
             # Analyze
             analysis = self.analyzer.analyze(context)

CriticalThinking/app/services/improvement_agent.py

@@ -4,11 +4,12 @@ from app.services.agent_orchestrator import BaseAgent
 class ImprovementAgent(BaseAgent):
     def generate_improvements(self, weaknesses: List[str]) -> Dict[str, Any]:
         system_prompt = """You are an AI research scientist and senior architect.
-Your goal is to generate impactful and creative ideas for improving a codebase.
+Your goal is to generate impactful and creative ideas for improving a codebase or designing a new one.
 Consider:
-- Refactoring for better scalability.
-- Replacing custom implementations with state-of-the-art Hugging Face models or open-source projects.
-- Improving performance and maintainability."""
+- Decomposing the project into independent, modular components.
+- Suggesting state-of-the-art Hugging Face models (Spaces/Models) or GitHub projects to serve as functional components.
+- Ensuring components communicate via FastAPI endpoints.
+- Replacing custom implementations with existing open-source projects to reduce maintenance."""
         user_prompt = f"Given these weaknesses:\n{weaknesses}\n\nPropose a next-step improvement roadmap. Respond in JSON with format:\n{{\n  'improvements': [\n    {{\n      'weakness': 'the identified weakness',\n      'proposal': 'detailed improvement plan',\n      'replacement_search_query': 'query for Hugging Face or GitHub',\n      'interestingness': 1-10,\n      'feasibility': 1-10\n    }}\n  ]\n}}"
         return self._get_response(system_prompt, user_prompt, response_format={"type": "json_object"})
 

README.md

@@ -7,7 +7,6 @@ sdk: docker
 pinned: false
 app_port: 7860
 ---
-
 <h1 align="center">
   <a href="https://github.com/SakanaAI/AI-Scientist/blob/main/docs/logo_2.png">
     <img src="docs/logo_2.png" width="215" /></a><br>

hf_app.py

@@ -16,7 +16,7 @@ with gr.Blocks(title="Critical Code Agent") as demo:
     gr.Markdown("Autonomous agent system for deep architectural analysis and software weakness identification.")
     
     with gr.Row():
-        repo_url = gr.Textbox(label="Repository URL", placeholder="https://github.com/username/repo")
+        repo_url = gr.Textbox(label="Repository URL (Optional)", placeholder="https://github.com/username/repo")
         project_desc = gr.Textbox(label="Project Description", placeholder="Brief description of the project")
     
     analyze_btn = gr.Button("Analyze Repository", variant="primary")