Update api.py

api.py

@@ -1,12 +1,13 @@
 import os
 import aiohttp
 from dotenv import load_dotenv
+from urllib.parse import urlencode
 
 load_dotenv()
 
 async def query_rag_api(question):
     """
-    Asynchronously query the RAG API for answer and contexts.
+    Asynchronously query the RAG model via Streamlit endpoint.
 
     Args:
         question (str): The user's question.
@@ -15,27 +16,17 @@ async def query_rag_api(question):
         dict: Answer, contexts, or error message.
 
     Note:
-        Replace API_ENDPOINT with the actual endpoint of your Hugging Face Space
-        (e.g., https://<your-space-id>.hf.space/ask) after deployment.
+        Replace API_ENDPOINT with your Space's URL (e.g., https://<your-space-id>.hf.space)
+        after deployment.
     """
-    # Placeholder - REPLACE with your Space's endpoint
-    API_ENDPOINT = "https://huggingface.co/spaces/samim2024/testing:8000/ask"  # Update to Space URL post-deployment
-    api_key = os.getenv("HUGGINGFACEHUB_API_TOKEN")
-    
-    if not api_key:
-        return {"error": "HUGGINGFACEHUB_API_TOKEN not set", "answer": "", "contexts": []}
-
-    headers = {"Content-Type": "application/json"}
-    payload = {"question": question}
+    # Placeholder - REPLACE with your Space's URL
+    API_ENDPOINT = "https://huggingface.co/spaces/samim2024/testing"
+    params = {"query": question}
+    url = f"{API_ENDPOINT}?{urlencode(params)}"
 
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.post(
-                API_ENDPOINT,
-                json=payload,
-                headers=headers,
-                timeout=10
-            ) as response:
+            async with session.get(url, timeout=10) as response:
                 response_dict = await response.json()
                 if response_dict.get("error"):
                     return {"error": response_dict["error"], "answer": "", "contexts": []}
@@ -44,4 +35,4 @@ async def query_rag_api(question):
                     "contexts": response_dict.get("contexts", [])
                 }
     except Exception as e:
-        return {"error": f"API call failed: {str(e)}", "answer": "", "contexts": []}
+        return {"error": f"Query failed: {str(e)}", "answer": "", "contexts": []}