Update app.py

app.py

@@ -38,7 +38,7 @@ def query_vectara(question):
                 "rerankingConfig": {
                     "rerankerId": 272725718,
                     "mmrConfig": {
-                        "diversityBias": 0.3
+                        "diversityBias": 0.27
                     }
                 },
                 "corpusKey": [
@@ -72,13 +72,16 @@ def query_vectara(question):
         headers=api_key_header
     )
 
+
     if response.status_code == 200:
         query_data = response.json()
-        response_message = f"Response from Vectara API: {json.dumps(query_data, indent=2)}"
+        # Extract summary and the first 5 sources
+        summary_text = query_data[0]["summary"]["text"]
+        sources = query_data[0]["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:
-        response_message = f"Error: {response.status_code}"
-
-    return response_message
+        return f"Error: {response.status_code}"
 
 iface = gr.Interface(
     fn=query_vectara,