Update app.py

app.py

@@ -7,7 +7,7 @@ import requests
 ragie_api_key = os.getenv('RAGIE_API_KEY')
 
 def get_chunks(query):
-    """Retrieve chunks with detailed API response checking"""
+    """Retrieve chunks using exact format from docs"""
     try:
         # Print API key status (safely)
         if ragie_api_key:
@@ -17,67 +17,50 @@ def get_chunks(query):
             
         print(f"\nSending query: {query}")
             
-        # Make the API call
+        # Make the API call exactly as shown in docs
         response = requests.post(
             "https://api.ragie.ai/retrievals",
             headers={
                 "Content-Type": "application/json",
-                "Authorization": f"Bearer {ragie_api_key}"
+                "Authorization": "Bearer " + ragie_api_key
             },
             json={
                 "query": query,
                 "filter": {
-                    "scope": "tutorial"  # Adjust this to match your scope
+                    "scope": "tutorial"
                 }
             }
         )
         
-        # Print full response details
+        # Print response status
         print(f"\nAPI Response Status: {response.status_code}")
-        print(f"Response Headers: {dict(response.headers)}")
         
         if response.status_code != 200:
-            error_detail = ""
-            try:
-                error_detail = response.json()
-                print(f"Error Response Body: {error_detail}")
-            except:
-                error_detail = response.text
-                print(f"Error Response Text: {error_detail}")
-            return f"""API Call Failed:
-Status Code: {response.status_code}
-Error Details: {error_detail}"""
+            return f"API Call Failed: Status Code {response.status_code}"
             
-        # Parse the successful response
+        # Parse the response
         data = response.json()
-        print(f"\nFound {len(data.get('scored_chunks', []))} chunks")
+        print(f"\nAPI Response: {json.dumps(data, indent=2)}")
         
-        # Format the chunks with API response info
-        result = f"""API Call Successful:
-Status Code: {response.status_code}
-Number of Chunks: {len(data.get('scored_chunks', []))}
-
-Retrieved Chunks:
-"""
+        # Format the chunks
+        result = "Retrieved Chunks:\n\n"
         for i, chunk in enumerate(data.get("scored_chunks", []), 1):
-            result += f"\nChunk {i} (Score: {chunk['score']}):\n"
+            result += f"Chunk {i}:\n"
             result += f"{chunk['text']}\n"
-            result += "-" * 50
+            result += "-" * 50 + "\n"
         
-        return result if data.get("scored_chunks") else "API call successful but no chunks were found for this query."
+        return result if data.get("scored_chunks") else "No chunks found for this query."
         
     except Exception as e:
-        error_msg = f"Error during API call: {str(e)}"
-        print(error_msg)
-        return error_msg
+        return f"Error: {str(e)}"
 
-# Create a simple interface
+# Create interface
 demo = gr.Interface(
     fn=get_chunks,
     inputs=gr.Textbox(label="Enter your query"),
-    outputs=gr.Textbox(label="API Response and Retrieved Chunks", lines=20),
-    title="Ragie Chunk Retriever with API Debugging",
-    description="Enter a query to see full API response details and retrieved chunks."
+    outputs=gr.Textbox(label="Retrieved Chunks", lines=20),
+    title="Ragie Chunk Retriever",
+    description="Enter a query to see what chunks Ragie retrieves."
 )
 
 if __name__ == "__main__":