Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 import requests
 import json
-import pprint
+import pandas as pd
 
 # Function to fetch data from Google Knowledge Graph
-def fetch_data(query, api_key):
-    url = f"https://kgsearch.googleapis.com/v1/entities:search?query={query}&key={api_key}&limit=1"
+def fetch_data(query, api_key, limit):
+    url = f"https://kgsearch.googleapis.com/v1/entities:search?query={query}&key={api_key}&limit={limit}"
     response = requests.get(url)
     json_data = json.loads(response.text)
     return json_data
@@ -21,20 +21,26 @@ if not api_key:
     api_key = st.text_input('Enter your Google API Key:', type="password")
 
 search_query = st.text_input('Enter Search Query:', 'Python Programming')
+output_count = st.slider('Number of Results:', min_value=1, max_value=10, value=1)
 
 if st.button('Search'):
     if api_key:
-        result = fetch_data(search_query, api_key)
+        result = fetch_data(search_query, api_key, output_count)
         try:
-            item = result['itemListElement'][0]['result']
-            st.subheader(f"Name: {item['name']}")
-            st.write(f"Description: {item.get('description', 'N/A')}")
-            st.write(f"Detail: {item.get('detailedDescription', {}).get('articleBody', 'N/A')}")
-            st.write(f"URL: {item.get('url', 'N/A')}")
-            
-            # Pretty-print the entire JSON response to display all outputs
-            st.subheader("Full JSON Response")
-            st.write(pprint.pformat(result))
+            items = result['itemListElement']
+            data = []
+            for item in items:
+                result_dict = {}
+                result_item = item['result']
+                result_dict['Name'] = result_item.get('name', 'N/A')
+                result_dict['Description'] = result_item.get('description', 'N/A')
+                result_dict['Detail'] = result_item.get('detailedDescription', {}).get('articleBody', 'N/A')
+                result_dict['URL'] = result_item.get('url', 'N/A')
+                data.append(result_dict)
+                
+            # Convert list of dictionaries to DataFrame for table display
+            df = pd.DataFrame(data)
+            st.table(df)
         except:
             st.write('No results found.')
     else: