Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import requests
 import json
+import logging
+
+logging.basicConfig(level=logging.DEBUG)
 
 API_URL = "https://sendthat.cc"
 
@@ -11,30 +14,50 @@ def list_indexes():
         response.raise_for_status()
         indexes = response.json()
         
-        # Check if the response is just ["indexes"] and replace with actual index names
-        if indexes == ["indexes"]:
-            return ["history", "medical"]  # Hardcoded index names
-        elif isinstance(indexes, list) and indexes:
-            return indexes
+        logging.debug(f"API response for indexes: {indexes}")
+        
+        if isinstance(indexes, list):
+            if "indexes" in indexes:
+                return ["history", "medical"]  # Hardcoded index names
+            elif indexes:
+                return indexes
+            else:
+                return ["No indexes available"]
+        elif isinstance(indexes, dict) and "indexes" in indexes:
+            return indexes["indexes"] if indexes["indexes"] else ["No indexes available"]
         else:
-            return ["No indexes available"]
+            logging.warning(f"Unexpected response format: {indexes}")
+            return ["history", "medical"]  # Fallback to hardcoded index names
     except requests.exceptions.RequestException as e:
-        print(f"Error fetching indexes: {e}")
+        logging.error(f"Error fetching indexes: {e}")
         return ["Error fetching indexes"]
+    except json.JSONDecodeError as e:
+        logging.error(f"Error decoding JSON: {e}")
+        return ["Error decoding response"]
 
 def search_document(index_name, query, k):
-    url = f"{API_URL}/search/{index_name}"
-    payload = {"text": query, "k": k}
-    headers = {"Content-Type": "application/json"}
-    response = requests.post(url, json=payload, headers=headers)
-    return response.json(), "", k  # Return empty string to clear input
+    try:
+        url = f"{API_URL}/search/{index_name}"
+        payload = {"text": query, "k": k}
+        headers = {"Content-Type": "application/json"}
+        response = requests.post(url, json=payload, headers=headers)
+        response.raise_for_status()
+        return response.json(), "", k
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error in search: {e}")
+        return {"error": str(e)}, query, k
 
 def qa_document(index_name, question, k):
-    url = f"{API_URL}/qa/{index_name}"
-    payload = {"text": question, "k": k}
-    headers = {"Content-Type": "application/json"}
-    response = requests.post(url, json=payload, headers=headers)
-    return response.json(), "", k  # Return empty string to clear input
+    try:
+        url = f"{API_URL}/qa/{index_name}"
+        payload = {"text": question, "k": k}
+        headers = {"Content-Type": "application/json"}
+        response = requests.post(url, json=payload, headers=headers)
+        response.raise_for_status()
+        return response.json(), "", k
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error in QA: {e}")
+        return {"error": str(e)}, question, k
 
 def refresh_indexes():
     indexes = list_indexes()