Upload folder using huggingface_hub

deep_research.py

@@ -4,6 +4,16 @@ from research_manager import ResearchManager, ResearchManagerAgent
 from agents import Runner, trace, gen_trace_id
 import os
 
+# Import spaces for ZeroGPU compatibility
+try:
+    import spaces
+except ImportError:
+    # Fallback for local development
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            return func
+
 load_dotenv(override=True)
 
 # Available models for user selection
@@ -17,6 +27,7 @@ AVAILABLE_MODELS = [
     "o1-mini"
 ]
 
+@spaces.GPU
 async def handle_query_submission(query: str, current_state: dict, api_key: str, model: str):
     """Handle initial query submission - generate clarifying questions with progress"""
     if not query.strip():
@@ -50,6 +61,7 @@ async def handle_query_submission(query: str, current_state: dict, api_key: str,
     except Exception as e:
         return f"❌ Error generating clarifying questions: {str(e)}", gr.update(visible=False), gr.update(visible=False), current_state
 
+@spaces.GPU
 async def handle_research_with_answers(answers: str, current_state: dict, email_address: str, send_email: bool):
     """Handle research execution with clarification answers with progress updates"""
     if not current_state.get("query"):
@@ -139,6 +151,7 @@ Please use these clarifications to focus and refine the research approach."""
     except Exception as e:
         return f"❌ Error during research: {str(e)}", current_state
 
+@spaces.GPU
 async def run_direct_research(query: str, api_key: str, model: str, email_address: str = "", send_email: bool = False):
     """Run research directly without clarification using the new agent-based system"""
     if not query.strip():
@@ -198,6 +211,7 @@ async def run_direct_research(query: str, api_key: str, model: str, email_addres
         print(f"Error details: {error_details}")
         return f"❌ Error during research: {str(e)}\n\nPlease check your API key and model selection, or try the Legacy Quick Research option if this persists."
 
+@spaces.GPU
 async def run_legacy_research(query: str, api_key: str, model: str, email_address: str, send_email: bool):
     """Run research using the original ResearchManager class with email options"""
     if not query.strip():
@@ -253,6 +267,7 @@ async def run_legacy_research(query: str, api_key: str, model: str, email_addres
         print(f"Error details: {error_details}")
         return f"❌ Error during research: {str(e)}\n\nPlease check your API key and model selection."
 
+@spaces.GPU
 async def run_direct_research_with_progress(query: str, api_key: str, model: str, email_address: str = "", send_email: bool = False):
     """Run direct research with real-time progress tracking"""
     if not query.strip():
@@ -283,6 +298,7 @@ async def run_direct_research_with_progress(query: str, api_key: str, model: str
         print(f"Error details: {error_details}")
         yield f"❌ Error during research: {str(e)}\n\nPlease check your API key and model selection."
 
+@spaces.GPU
 async def run_enhanced_research_with_progress(query: str, api_key: str, model: str, email_address: str = "", send_email: bool = False):
     """Run enhanced research with real-time progress tracking"""
     if not query.strip():
@@ -313,6 +329,7 @@ async def run_enhanced_research_with_progress(query: str, api_key: str, model: s
         print(f"Error details: {error_details}")
         yield f"❌ Error during enhanced research: {str(e)}\n\nPlease check your API key and model selection."
 
+@spaces.GPU
 async def run_clarified_research_with_progress(answers: str, current_state: dict, email_address: str, send_email: bool):
     """Run research with clarification answers and real-time progress tracking"""
     if not current_state.get("query"):

requirements.txt

@@ -7,4 +7,5 @@ requests>=2.32.3
 bs4>=0.0.2
 httpx>=0.28.1
 pydantic>=2.0.0
-typing-extensions>=4.0.0 
+typing-extensions>=4.0.0
+spaces>=0.28.3