Update app.py

app.py

@@ -72,19 +72,35 @@ def query_vectara(question):
         verify=True,
         headers=api_key_header
     )
-
-
+  
     if response.status_code == 200:  
         query_data = response.json()  
-        print(query_data)    
+        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)}"  
+            sources_info = []  
+              
+            # Iterate over all response sets  
+            for response_set in query_data.get('responseSet', []):  
+                # Extract sources  
+                for source in response_set.get('response', []):  
+                    source_metadata = source.get('metadata', [])  
+                    source_info = {}  
+  
+                    for metadata in source_metadata:  
+                        metadata_name = metadata.get('name', '')  
+                        metadata_value = metadata.get('value', '')  
+  
+                        if metadata_name == 'title':  
+                            source_info['title'] = metadata_value  
+                        elif metadata_name == 'author':  
+                            source_info['author'] = metadata_value  
+                        elif metadata_name == 'pageNumber':  
+                            source_info['page number'] = metadata_value  
+  
+                    if source_info:  
+                        sources_info.append(source_info)  
+              
+            return f"Sources:\n{json.dumps(sources_info, indent=2)}"  
         else:  
             return "No data found in the response."  
     else: