Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,15 +1,14 @@
 import streamlit as st
 from PIL import Image
-import os
 import wikipedia
 import csv
 from datetime import datetime
 import tempfile
+import os
 
 # ====== Setup ======
 USERS = {}
 
-# Use a writable temporary directory to avoid permission issues
 UPLOAD_FOLDER = tempfile.mkdtemp()
 MEDIA_FOLDER = os.path.join(UPLOAD_FOLDER, "feedback_media")
 FEEDBACK_FILE = os.path.join(UPLOAD_FOLDER, "feedback.csv")
@@ -40,7 +39,6 @@ for key, default in {
         st.session_state[key] = default
 
 # ====== Helper Functions ======
-
 def send_otp(email):
     st.session_state.otp_sent_to = email
     st.session_state.otp_verified = False
@@ -71,15 +69,12 @@ def fetch_wikipedia_summary(title, lang="en"):
         return f"Error fetching summary: {str(e)}"
 
 def save_temp_file(uploaded_file, filetype):
-    if uploaded_file is not None:
+    if uploaded_file:
         filename = f"{filetype}_{datetime.now().strftime('%Y%m%d%H%M%S')}_{uploaded_file.name}"
         filepath = os.path.join(MEDIA_FOLDER, filename)
-        try:
-            with open(filepath, "wb") as f:
-                f.write(uploaded_file.getbuffer())
-            return filepath
-        except Exception as e:
-            st.error(f"Failed to save {filetype} file: {e}")
+        with open(filepath, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        return filepath
     return ""
 
 def save_feedback(email, place, state, district, text, img_path="", audio_path="", video_path=""):
@@ -90,6 +85,7 @@ def save_feedback(email, place, state, district, text, img_path="", audio_path="
                 "user_email", "place_name", "state", "district", "feedback_text",
                 "image_file", "audio_file", "video_file", "timestamp"
             ])
+
     with open(FEEDBACK_FILE, 'a', newline='', encoding='utf-8') as f:
         writer = csv.writer(f)
         writer.writerow([
@@ -98,7 +94,6 @@ def save_feedback(email, place, state, district, text, img_path="", audio_path="
         ])
 
 # ====== Pages ======
-
 def signup_page():
     st.header("🚀 Signup")
     name = st.text_input("Name")
@@ -112,7 +107,7 @@ def signup_page():
         else:
             USERS[email] = {"name": name, "email": email, "contact": contact}
             send_otp(email)
-            st.success(f"User {name} registered! OTP sent to {email} (simulated).")
+            st.success(f"User {name} registered! OTP sent to {email} (simulated). Please verify OTP on Login page.")
             st.session_state.login_email = email
 
 def login_page():
@@ -121,15 +116,15 @@ def login_page():
     if st.button("Send OTP"):
         if email in USERS:
             send_otp(email)
-            st.success(f"OTP sent to {email} (simulated).")
+            st.success(f"OTP sent to {email} (simulated). Please enter OTP below.")
             st.session_state.login_email = email
         else:
             st.error("Email not found. Please sign up first.")
     if st.session_state.login_email == email:
-        otp = st.text_input("Enter OTP (use 1234)", type="password")
+        otp = st.text_input("Enter OTP (use 1234 for demo)", type="password")
         if st.button("Verify OTP"):
             if verify_otp(otp):
-                st.success("Logged in successfully.")
+                st.success("OTP verified! Logged in successfully.")
             else:
                 st.error("Incorrect OTP or no OTP sent.")
 
@@ -138,52 +133,61 @@ def main_app():
     if st.sidebar.button("Logout"):
         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", height=100)
-    feedback_image = st.sidebar.file_uploader("Upload image", type=["jpg","jpeg","png"])
-    feedback_audio = st.sidebar.file_uploader("Upload audio", type=["mp3","wav","m4a"])
-    feedback_video = st.sidebar.file_uploader("Upload video", type=["mp4","mov","avi"])
-
-    if feedback_image: st.sidebar.image(feedback_image, caption="Preview", use_container_width=True)
-    if feedback_audio: st.sidebar.audio(feedback_audio)
-    if feedback_video: st.sidebar.video(feedback_video)
+    feedback_text = st.sidebar.text_area("Your thoughts about this place", height=100)
+    feedback_image = st.sidebar.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+    feedback_audio = st.sidebar.file_uploader("Upload audio", type=["mp3", "wav", "m4a"])
+    feedback_video = st.sidebar.file_uploader("Upload video", type=["mp4", "mov", "avi"])
+
+    if feedback_image:
+        st.sidebar.image(feedback_image, caption="Your uploaded image", use_container_width=True)
+    if feedback_audio:
+        st.sidebar.audio(feedback_audio, format=feedback_audio.type)
+    if feedback_video:
+        st.sidebar.video(feedback_video)
 
     st.title("📚 Gyana Vedika - Cultural Explorer")
     state = st.selectbox("Select State", list(STATES.keys()))
     district = st.selectbox("Select District", STATES[state])
     st.subheader(f"Explore {district}, {state}")
 
-    uploaded_image = st.file_uploader("Upload cultural place image", type=["png","jpg","jpeg"])
+    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, use_container_width=True)
-        fn = uploaded_image.name.lower()
-        keywords = {
-            "charminar":"Charminar","golconda":"Golconda Fort","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"
+        st.image(uploaded_image, caption="Uploaded Image", use_container_width=True)
+        filename = uploaded_image.name.lower()
+        keywords_map = {
+            "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.items():
-            if kw in fn:
+        for kw, place in keywords_map.items():
+            if kw in filename:
                 place_to_search = place
-                st.success(f"Recognized: {place}")
+                st.success(f"Recognized Place: {place_to_search}")
                 break
         if not place_to_search:
-            st.warning("Could not recognize place. Rename your file.")
+            st.warning("Could not recognize place. Try renaming the image file with landmark name.")
 
-    manual_place = st.text_input("Or enter place name manually")
+    manual_place = st.text_input("Or manually enter place name to search info")
     if manual_place.strip():
         place_to_search = manual_place.strip()
 
     if place_to_search:
-        lang = st.selectbox("Wikipedia language", ["English","Hindi","Telugu","Tamil","Marathi","Bengali"])
-        codes = {"English":"en","Hindi":"hi","Telugu":"te","Tamil":"ta","Marathi":"mr","Bengali":"bn"}
-        if st.button(f"Get info: {place_to_search}"):
-            with st.spinner("Fetching..."):
-                summary = fetch_wikipedia_summary(place_to_search, lang=codes[lang])
-                st.markdown(f"### {place_to_search} ({lang})")
+        language = st.selectbox("Select Wikipedia summary language", ["English", "Hindi", "Telugu", "Tamil", "Marathi", "Bengali"])
+        lang_codes = {"English": "en", "Hindi": "hi", "Telugu": "te", "Tamil": "ta", "Marathi": "mr", "Bengali": "bn"}
+        lang_code = lang_codes.get(language, "en")
+        if st.button(f"Get info about {place_to_search}"):
+            with st.spinner(f"Fetching Wikipedia summary for {place_to_search}..."):
+                summary = fetch_wikipedia_summary(place_to_search, lang=lang_code)
+                st.markdown(f"### Summary of {place_to_search}")
                 st.write(summary)
 
     if st.sidebar.button("Submit Feedback"):
@@ -191,24 +195,42 @@ def main_app():
             img_path = save_temp_file(feedback_image, "img")
             audio_path = save_temp_file(feedback_audio, "audio")
             video_path = save_temp_file(feedback_video, "video")
-            st.sidebar.write(f"Saved files:\n• Image: {img_path}\n• Audio: {audio_path}\n• Video: {video_path}")
-            save_feedback(st.session_state.email, place_to_search or "Unknown", state, district, feedback_text.strip(), img_path, audio_path, video_path)
-            st.sidebar.success("Thanks for your feedback!")
+            save_feedback(
+                email=st.session_state.email,
+                place=place_to_search or "Unknown",
+                state=state,
+                district=district,
+                text=feedback_text.strip(),
+                img_path=img_path,
+                audio_path=audio_path,
+                video_path=video_path
+            )
+            st.sidebar.success("Thank you! Feedback submitted.")
         else:
-            st.sidebar.warning("Please enter feedback text.")
+            st.sidebar.warning("Please enter your feedback text.")
 
 # ====== UI Styling ======
 st.set_page_config(page_title="Gyana Vedika", layout="wide")
+
 st.markdown("""
 <style>
-.stButton>button { background-color: #008080; color: white; font-weight: bold; border-radius: 8px; }
-.stTextInput>div>input, .stSelectbox>div>div>select, .stTextArea>div>textarea { border:2px solid #008080!important; border-radius:6px!important; }
-h1, h2, h3, h4 { color:#004d4d; }
-</style>""", unsafe_allow_html=True)
+.stButton>button {
+    background-color: #008080; color: white; font-weight: bold; border-radius: 10px;
+}
+.stTextInput>div>input, .stSelectbox>div>div>select, .stTextArea>div>textarea {
+    border: 2px solid #008080 !important; border-radius: 8px !important;
+}
+h1, h2, h3, h4, h5, h6 {
+    color: #004d4d;
+}
+</style>
+""", unsafe_allow_html=True)
 
-# ====== Run App ======
 if not st.session_state.logged_in:
-    page = st.sidebar.radio("Choose Option", ["Signup","Login"])
-    signup_page() if page=="Signup" else login_page()
+    page = st.sidebar.radio("Choose Option", ["Signup", "Login"])
+    if page == "Signup":
+        signup_page()
+    elif page == "Login":
+        login_page()
 else:
     main_app()