Update app.py

app.py

@@ -73,18 +73,22 @@ def query_vectara(question):
     )
 
 
-    if response.status_code == 200:
-        query_data = response.json()
-        if query_data:
-            # Extract summary and the first 5 sources
-            summary_text = query_data["summary"]["text"]
-            sources = query_data["responseSet"]["responseList"][:5]
-            sources_text = [source["text"] for source in sources]
-            return f"Summary: {summary_text}\n\nSources:\n{json.dumps(sources_text, indent=2)}"
-        else:
-            return "No data found in the response."
-    else:
-        return f"Error: {response.status_code}"
+    if response.status_code == 200:  
+        query_data = response.json()  
+        print(query_data)    
+        if query_data:  
+            # Extract summary and the first 5 sources  
+            response_set = query_data.get('responseSet', [{}])[0]  # get the first response set  
+            summary = response_set.get('summary', [{}])[0]  # get the first summary  
+            summary_text = summary.get('text', 'No summary available')  
+            sources = response_set.get('response', [])[:5]  
+            sources_text = [source.get('text', '') for source in sources]  
+            return f"Summary: {summary_text}\n\nSources:\n{json.dumps(sources_text, indent=2)}"  
+        else:  
+            return "No data found in the response."  
+    else:  
+        return f"Error: {response.status_code}"  
+
 
 iface = gr.Interface(
     fn=query_vectara,