Update app.py

app.py

@@ -14,21 +14,27 @@ HEADERS = {
 }
 
 # --- Core API Functions ---
-def query_rag_system(query: str, max_cards: int = 10) -> dict:
+def query_rag_system(query: str, max_cards: int = 10) -> dict: # Default changed to 10
     """Query the RAG system with a historical question."""
     try:
         payload = {"query": query, "max_cards": max_cards, "include_sources": True}
         response = requests.post(f"{API_URL}/query", json=payload, headers=HEADERS)
+        response.raise_for_status() # Raise an error for bad status codes
         return response.json()
+    except requests.exceptions.RequestException as e:
+        return {"error": f"Network Error: {e}"}
     except Exception as e:
         return {"error": str(e)}
 
-def search_cards_semantic(query: str, max_cards: int = 10) -> dict:
+def search_cards_semantic(query: str, max_cards: int = 10) -> dict: # Default changed to 10
     """Perform semantic search on cards."""
     try:
         payload = {"query": query, "max_cards": max_cards}
         response = requests.post(f"{API_URL}/search", json=payload, headers=HEADERS)
+        response.raise_for_status() # Raise an error for bad status codes
         return response.json()
+    except requests.exceptions.RequestException as e:
+        return {"error": f"Network Error: {e}"}
     except Exception as e:
         return {"error": str(e)}
 
@@ -48,10 +54,9 @@ def format_rag_response(result: dict) -> str:
     
     if sources:
         response += f"**📚 Sources ({len(sources)} documents):**\n"
-        for i, source in enumerate(sources[:3], 1):  # Limit to top 3 sources
+        # MODIFICATION: Loop through all sources instead of just the top 3
+        for i, source in enumerate(sources, 1):
             response += f"{i}. {source}\n"
-        if len(sources) > 3:
-            response += f"... and {len(sources) - 3} more sources\n"
         response += "\n"
     
     if cards_used:
@@ -71,7 +76,8 @@ def format_search_response(result: dict) -> str:
     
     response = f"🔍 **Found {len(cards)} relevant historical entries:**\n\n"
     
-    for i, card in enumerate(cards[:3], 1):  # Show top 3 results
+    # MODIFICATION: Show all results instead of just the top 3
+    for i, card in enumerate(cards, 1):
         title = card.get('title', 'Untitled')
         summary = card.get('summary', 'No summary available')
         relevance = card.get('relevance_score', 0)
@@ -82,9 +88,6 @@ def format_search_response(result: dict) -> str:
         
         response += f"**{i}. {title}** (Relevance: {relevance:.3f})\n{summary}\n\n"
     
-    if len(cards) > 3:
-        response += f"*... and {len(cards) - 3} more results*\n"
-    
     processing_time = result.get("processing_time", 0)
     response += f"⏱️ Search completed in {processing_time:.2f}s"
     
@@ -95,8 +98,8 @@ def chatbot_respond(message: str, history: List[Tuple[str, str]], search_mode: b
     if not message.strip():
         return "", history
     
-    # Determine max cards based on query complexity
-    max_cards = 10 if len(message.split()) < 5 else 10
+    # MODIFICATION: Set max_cards to 10 for both modes
+    max_cards = 10
     
     if search_mode:
         # Use semantic search for browsing/exploring
@@ -202,14 +205,17 @@ def create_interface():
 if __name__ == "__main__":
     print("🚀 Launching Historical Knowledge Assistant...")
     print(f"🌐 API URL: {API_URL}")
-    print(f"🔑 Using API token: {API_TOKEN[:10]}...")
-    
+    if API_TOKEN:
+        print(f"🔑 Using API token: {API_TOKEN[:4]}...{API_TOKEN[-4:]}")
+    else:
+        print("🔑 API token not found. Please set the API_TOKEN secret in your Space settings.")
+
     demo = create_interface()
     
     # For Hugging Face Spaces deployment
     demo.launch(
-        share=False,  # Set to True for temporary sharing
-        server_name="0.0.0.0",  # Required for HF Spaces
-        server_port=7860,       # Standard port for HF Spaces
+        share=False,
+        server_name="0.0.0.0",
+        server_port=7860,
         show_error=True
     )