Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,12 +7,10 @@ from datetime import datetime
 
 # ====== Setup ======
 USERS = {}
-# Use /tmp directory for Hugging Face Spaces, which is writable
 UPLOAD_FOLDER = "/tmp/uploads"
 MEDIA_FOLDER = os.path.join(UPLOAD_FOLDER, "feedback_media")
 FEEDBACK_FILE = os.path.join(UPLOAD_FOLDER, "feedback.csv")
 
-# Create directories if they don't exist
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(MEDIA_FOLDER, exist_ok=True)
 
@@ -28,13 +26,13 @@ STATES = {
     "Andhra Pradesh": ["Visakhapatnam", "Vijayawada", "Guntur", "Tirupati", "Kurnool"],
 }
 
-# ====== Session State Initialization ======
 for key, default in {
     "logged_in": False,
     "email": "",
     "otp_sent_to": None,
     "otp_verified": False,
     "login_email": None,
+    "place_image": None,
 }.items():
     if key not in st.session_state:
         st.session_state[key] = default
@@ -56,7 +54,7 @@ def verify_otp(entered_otp):
     return False
 
 def logout():
-    for key in ["logged_in", "email", "otp_sent_to", "otp_verified", "login_email"]:
+    for key in ["logged_in", "email", "otp_sent_to", "otp_verified", "login_email", "place_image"]:
         st.session_state[key] = False if isinstance(st.session_state.get(key), bool) else None
 
 def fetch_wikipedia_summary(title, lang="en"):
@@ -137,7 +135,6 @@ def main_app():
         logout()
         st.experimental_rerun()
 
-    # Feedback inputs on left sidebar
     st.sidebar.markdown("---")
     st.sidebar.subheader("📝 Give Feedback")
     feedback_text = st.sidebar.text_area("Your thoughts about this place", height=100)
@@ -145,7 +142,6 @@ def main_app():
     feedback_audio = st.sidebar.file_uploader("Upload audio", type=["mp3", "wav", "m4a"])
     feedback_video = st.sidebar.file_uploader("Upload video", type=["mp4", "mov", "avi"])
 
-    # Preview uploaded feedback files
     if feedback_image:
         st.sidebar.image(feedback_image, caption="Your uploaded image", use_container_width=True)
     if feedback_audio:
@@ -153,7 +149,6 @@ def main_app():
     if feedback_video:
         st.sidebar.video(feedback_video)
 
-    # Main content
     st.title("📚 Gyana Vedika - Cultural Explorer")
 
     state = st.selectbox("Select State", list(STATES.keys()))
@@ -161,35 +156,22 @@ def main_app():
     st.subheader(f"Explore {district}, {state}")
 
     uploaded_image = st.file_uploader("Upload a cultural/historical place image", type=["png", "jpg", "jpeg"])
-    place_to_search = None
     if uploaded_image:
-        st.image(uploaded_image, caption="Uploaded Image", use_container_width=True)
-        filename = uploaded_image.name.lower()
+        st.session_state["place_image"] = uploaded_image
+
+    place_to_search = None
+
+    if st.session_state["place_image"]:
+        st.image(st.session_state["place_image"], caption="Uploaded Image", use_container_width=True)
+        filename = st.session_state["place_image"].name.lower()
         keywords_map = {
-            # Telangana Places
-            "charminar": "Charminar",
-            "golconda": "Golconda Fort",
-            "qutubshahi": "Qutb Shahi Tombs",
-            "birla": "Birla Mandir",
-            "salarjung": "Salar Jung Museum",
-            "warangal": "Warangal Fort",
-            "ramappa": "Ramappa Temple",
-            "bhadrakali": "Bhadra Kali Temple",
-            "kakatiya": "Kakatiya Kala Thoranam",
-            "pakhal": "Pakhal Lake",
-            "medak": "Medak Cathedral",
-            "nagarjuna": "Nagarjuna Sagar Dam",
-            # Other famous places (existing)
-            "taj": "Taj Mahal",
-            "gateway": "Gateway of India",
-            "qutub": "Qutub Minar",
-            "mysore": "Mysore Palace",
-            "hampi": "Hampi",
-            "konark": "Konark Sun Temple",
-            "varanasi": "Varanasi",
-            "madurai": "Meenakshi Temple",
-            "ajanta": "Ajanta Caves",
-            "ellora": "Ellora Caves",
+            "charminar": "Charminar", "golconda": "Golconda Fort", "qutubshahi": "Qutb Shahi Tombs",
+            "birla": "Birla Mandir", "salarjung": "Salar Jung Museum", "warangal": "Warangal Fort",
+            "ramappa": "Ramappa Temple", "bhadrakali": "Bhadra Kali Temple", "kakatiya": "Kakatiya Kala Thoranam",
+            "pakhal": "Pakhal Lake", "medak": "Medak Cathedral", "nagarjuna": "Nagarjuna Sagar Dam",
+            "taj": "Taj Mahal", "gateway": "Gateway of India", "qutub": "Qutub Minar", "mysore": "Mysore Palace",
+            "hampi": "Hampi", "konark": "Konark Sun Temple", "varanasi": "Varanasi", "madurai": "Meenakshi Temple",
+            "ajanta": "Ajanta Caves", "ellora": "Ellora Caves",
         }
         for kw, place in keywords_map.items():
             if kw in filename:
@@ -235,7 +217,6 @@ def main_app():
 
 # ====== UI Styling ======
 st.set_page_config(page_title="Gyana Vedika", layout="wide")
-
 st.markdown("""
 <style>
 .stButton>button {
@@ -250,6 +231,7 @@ h1, h2, h3, h4, h5, h6 {
 </style>
 """, unsafe_allow_html=True)
 
+# ====== Main Routing ======
 if not st.session_state.logged_in:
     page = st.sidebar.radio("Choose Option", ["Signup", "Login"])
     if page == "Signup":
@@ -257,4 +239,4 @@ if not st.session_state.logged_in:
     elif page == "Login":
         login_page()
 else:
-    main_app()
+    main_app()