fix

main.py

@@ -12,6 +12,7 @@ from pathlib import Path
 from datetime import datetime
 import time
 import random
+from collections import defaultdict, deque
 
 from content_analyzer.document_parser import DocumentProcessor
 from search_engine.indexer import RetrieverBuilder
@@ -19,6 +20,22 @@ from intelligence.orchestrator import AgentWorkflow
 from configuration import definitions, parameters
                                  
 
+# Rate limiting configuration - 3 requests per 60 seconds per IP
+WINDOW_S = 3600
+MAX_CALLS = 3
+_calls = defaultdict(deque)  # ip -> timestamps
+
+def rate_limit(request):
+    ip = getattr(request.client, "host", "unknown")
+    now = time.time()
+    q = _calls[ip]
+    while q and (now - q[0]) > WINDOW_S:
+        q.popleft()
+    if len(q) >= MAX_CALLS:
+        raise gr.Error(f"Rate limit: {MAX_CALLS} requests per {WINDOW_S}s. Please wait.")
+    q.append(now)
+
+
 # Example data for demo
 EXAMPLES = {
     "Generative AI and Jobs": {
@@ -602,12 +619,16 @@ window.addEventListener("load", tick);
 setInterval(tick, 500);
 '''
 
-    with gr.Blocks(theme=gr.themes.Soft(), title="SmartDoc AI") as demo:
+    # Launch server - Compatible with both local and Hugging Face Spaces
+    # HF Spaces sets SPACE_ID environment variable
+    is_hf_space = os.environ.get("SPACE_ID") is not None
+    
+    with gr.Blocks(title="SmartDoc AI") as demo:
         gr.Markdown("### SmartDoc AI - Document Q&A", elem_classes="app-title")
         gr.Markdown("Upload your documents and ask questions. Answers will appear below, just like a chat.", elem_classes="app-description")
         gr.Markdown("---")
 
-        # Examples dropdown
+        # Examples dropdown - visible for both local and HF Spaces
         example_dropdown = gr.Dropdown(
             label="Quick Start - Choose an Example",
             choices=list(EXAMPLES.keys()),
@@ -635,7 +656,8 @@ setInterval(tick, 500);
             "session_start": datetime.now().strftime("%Y-%m-%d %H:%M")
         })
 
-        def process_question(question_text, uploaded_files, chat_history):        
+        def process_question(question_text, uploaded_files, chat_history, request: gr.Request):
+            rate_limit(request)
             chat_history = chat_history or []            
             yield (
                 chat_history,
@@ -865,25 +887,53 @@ setInterval(tick, 500);
                 return [], "", "Select a valid example from the dropdown above"
             ex_data = EXAMPLES[example_key]
             question_text = ex_data["question"]
-            file_paths = ex_data["file_paths"]
-            import tempfile
-            temp_dir = tempfile.mkdtemp()
-            copied_files = []
-            file_info_text = f"Loaded: {example_key}\n\n"
-            for source_file_path in file_paths:
-                abs_source = os.path.abspath(source_file_path)
-                if os.path.exists(abs_source):
-                    filename = os.path.basename(abs_source)
-                    temp_file_path = os.path.join(temp_dir, filename)
-                    shutil.copy2(abs_source, temp_file_path)
-                    copied_files.append(temp_file_path)
-                    file_size_mb = os.path.getsize(temp_file_path) / (1024 * 1024)
-                    file_info_text += f"{filename} ({file_size_mb:.2f} MB)\n"
-                else:
-                    file_info_text += f"{source_file_path} not found\n"
-            if not copied_files:
-                return [], "", "Could not load example files"
-            return copied_files, question_text, file_info_text          
+            file_names = ex_data["file_paths"]
+            
+            # Try to download from HF dataset if on Spaces
+            if is_hf_space:
+                try:
+                    from huggingface_hub import hf_hub_download
+                    copied_files = []
+                    file_info_text = f"✅ Loaded: {example_key}\n\n"
+                    for file_path in file_names:
+                        filename = os.path.basename(file_path)
+                        try:
+                            local_path = hf_hub_download(
+                                repo_id="TilanTAB/smartdoc-samples",
+                                repo_type="dataset",
+                                filename=filename,
+                            )
+                            copied_files.append(local_path)
+                            file_size_mb = os.path.getsize(local_path) / (1024 * 1024)
+                            file_info_text += f"📄 {filename} ({file_size_mb:.2f} MB)\n"
+                        except Exception as e:
+                            logger.error(f"Failed to download {filename}: {e}")
+                            file_info_text += f"❌ {filename} - Download failed\n"
+                    if not copied_files:
+                        return [], "", "❌ Could not load example files from dataset"
+                    return copied_files, question_text, file_info_text
+                except ImportError:
+                    return [], "", "❌ huggingface_hub not installed"
+            else:
+                # Local mode - use files from samples directory
+                import tempfile
+                temp_dir = tempfile.mkdtemp()
+                copied_files = []
+                file_info_text = f"Loaded: {example_key}\n\n"
+                for source_file_path in file_names:
+                    abs_source = os.path.abspath(source_file_path)
+                    if os.path.exists(abs_source):
+                        filename = os.path.basename(abs_source)
+                        temp_file_path = os.path.join(temp_dir, filename)
+                        shutil.copy2(abs_source, temp_file_path)
+                        copied_files.append(temp_file_path)
+                        file_size_mb = os.path.getsize(temp_file_path) / (1024 * 1024)
+                        file_info_text += f"{filename} ({file_size_mb:.2f} MB)\n"
+                    else:
+                        file_info_text += f"{source_file_path} not found\n"
+                if not copied_files:
+                    return [], "", "Could not load example files"
+                return copied_files, question_text, file_info_text
         
         example_dropdown.change(
             fn=load_example,
@@ -905,7 +955,7 @@ setInterval(tick, 500);
         server_port = _find_open_port(configured_port)
         logger.info(f"Launching Gradio on port {server_port}")
         logger.info(f"Access the app at: http://127.0.0.1:{server_port}")
-        demo.launch(server_name="127.0.0.1", server_port=server_port, share=False)
+        demo.launch(theme=gr.themes.Soft(), server_name="127.0.0.1", server_port=server_port, share=False, css=css, js=js)
 
 
 if __name__ == "__main__":