Update app.py

app.py

@@ -151,19 +151,41 @@ with col6:
     st.button("✒️ Ink")
     st.button("📜 Scroll")
 
+
+
+
+def fetch_wikipedia_summary(keyword):
+    # Placeholder function for fetching Wikipedia summaries
+    # In a real app, you might use requests to fetch from the Wikipedia API
+    return f"Summary for {keyword}. For more information, visit Wikipedia."
+
+def create_search_url_wikipedia(keyword):
+    base_url = "https://www.wikipedia.org/search-redirect.php?family=wikipedia&language=en&search="
+    return base_url + keyword.replace(' ', '+')
+
+def create_search_url_google(keyword):
+    base_url = "https://www.google.com/search?q="
+    return base_url + keyword.replace(' ', '+')
+
 def display_images_and_wikipedia_summaries():
     st.title('Gallery with Related Stories')
     image_files = [f for f in os.listdir('.') if f.endswith('.png')]
     if not image_files:
         st.write("No PNG images found in the current directory.")
         return
+    
     for image_file in image_files:
         image = Image.open(image_file)
-        st.image(image, caption=image_file, use_column_width='always')
-        keyword = image_file.split('.')[0]  # Assumes keyword is the file name without extension
+        st.image(image, caption=image_file, use_column_width=True)
         
+        keyword = image_file.split('.')[0]  # Assumes keyword is the file name without extension
         wikipedia_summary = fetch_wikipedia_summary(keyword)
         st.write(wikipedia_summary)
+        
+        # Display Wikipedia and Google search links
+        wikipedia_url = create_search_url_wikipedia(keyword)
+        google_url = create_search_url_google(keyword)
+        st.markdown(f"[Wikipedia]({wikipedia_url}) | [Google]({google_url})")
 
 display_images_and_wikipedia_summaries()