Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,11 +3,11 @@ from PIL import Image
 import requests
 
 st.set_page_config(page_title="WikiExplorer AR", layout="centered")
-st.title("WikiExplorer AR (Streamlit Edition)")
+st.title("📷 WikiExplorer AR (Streamlit Edition)")
 
 # --- Multilingual language selector ---
 lang = st.selectbox(
-    "Select Language",
+    "🌐 Select Language",
     options=[
         ("English", "en"),
         ("हिन्दी", "hi"),
@@ -20,12 +20,12 @@ lang = st.selectbox(
 lang_code = lang[1]
 
 # --- Place name input ---
-place_name = st.text_input("Enter place name (e.g., Charminar)")
+place_name = st.text_input("🏛️ Enter place name (e.g., Charminar)")
 
 # --- Camera input ---
-img_file_buffer = st.camera_input("Take a picture (optional)")
+img_file_buffer = st.camera_input("📸 Take a picture (optional)")
 
-# --- Function: Wikipedia API ---
+# --- Function: Wikipedia + Commons API ---
 def get_place_info(place, lang):
     if not place:
         return None
@@ -36,11 +36,11 @@ def get_place_info(place, lang):
         wiki_resp = requests.get(wiki_url)
         wiki_data = wiki_resp.json() if wiki_resp.status_code == 200 else {}
 
-        # Wikimedia Commons images (example)
+        # Wikimedia Commons
         commons_url = (
             f"https://commons.wikimedia.org/w/api.php"
             f"?action=query&format=json&prop=imageinfo&generator=search"
-            f"&gsrsearch={place}&gsrlimit=5&iiprop=url"
+            f"&gsrsearch={place}&gsrlimit=3&iiprop=url"
         )
         commons_resp = requests.get(commons_url)
         commons_data = []
@@ -48,37 +48,47 @@ def get_place_info(place, lang):
             result = commons_resp.json().get('query', {}).get('pages', {})
             for page in result.values():
                 imginfo = page.get('imageinfo', [{}])[0]
-                commons_data.append({"url": imginfo.get('url')})
+                img_url = imginfo.get('url')
+                if img_url:
+                    commons_data.append({"url": img_url})
 
         return {
             "wikipedia": wiki_data,
             "commons": commons_data,
         }
     except Exception as e:
-        st.error(f"API request failed: {e}")
+        st.error(f"❌ API request failed: {e}")
         return None
 
-# --- Display info ---
+# --- Display place info ---
 if place_name.strip():
-    st.info(f"Fetching info for **{place_name}** in language **{lang_code}**...")
+    st.info(f"🔍 Fetching info for **{place_name}** in **{lang_code.upper()}**...")
     data = get_place_info(place_name, lang_code)
+
     if not data:
-        st.error("Could not get data. Please check the name or try again.")
+        st.error("⚠️ Could not retrieve data. Check the place name or try again.")
     else:
-        st.subheader(f"Information about {place_name}")
-        st.write(data['wikipedia'].get('extract', 'No info found.'))
+        st.subheader(f"📖 About {place_name}")
+        st.write(data['wikipedia'].get('extract', 'No information found.'))
 
         if data['commons']:
+            st.markdown("### 🖼️ Related Images")
             for img in data['commons']:
-                st.image(img['url'], width=250)
+                if img and img.get('url'):
+                    st.image(img['url'], width=250)
+        else:
+            st.warning("No images found on Wikimedia Commons.")
 
 # --- Show captured image ---
 if img_file_buffer is not None:
-    st.image(img_file_buffer, caption="Captured Image")
+    st.markdown("### 📷 Captured Image")
+    st.image(img_file_buffer, caption="Uploaded via camera", use_column_width=True)
 
+# --- Footer ---
 st.markdown("""
-- 📌 Supports text search + camera input.
+---
+- 📌 Supports text-based search and camera input.
 - 🌐 Multilingual Wikipedia summaries.
-- 🖼️ Wikimedia Commons images.
-- ✅ Runs entirely in Streamlit — no separate backend needed!
+- 🖼️ Integrated with Wikimedia Commons.
+- ✅ Runs entirely in Streamlit — ready for Hugging Face Spaces deployment!
 """)