Update app.py

app.py

@@ -1,19 +1,27 @@
-# # app.py
+# # # app.py
+
 # import streamlit as st
 # import requests
 # from datetime import datetime
 # from deep_translator import GoogleTranslator
 # from geopy.geocoders import Nominatim
-# from streamlit_lottie import st_lottie
 # from timezonefinder import TimezoneFinder
 # import pytz
 # from streamlit_js_eval import get_geolocation
 
-
-# # Set your Groq API Key
+# # ---------------------------
+# # API Key
+# # ---------------------------
 # GROQ_API_KEY = "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"
 
+# # ---------------------------
+# # Page Config
+# # ---------------------------
+# st.set_page_config(page_title="AI Disaster Assistant", layout="wide", initial_sidebar_state="expanded")
+
+# # ---------------------------
 # # Rescue numbers dictionary
+# # ---------------------------
 # rescue_contacts = {
 #     "Pakistan": {
 #         "Ambulance": "115",
@@ -35,30 +43,30 @@
 #     }
 # }
 
-# def get_local_time(lat, lon):
+# # ---------------------------
+# # Timezone Helper
+# # ---------------------------
+# def get_timezone(lat, lon):
 #     tf = TimezoneFinder()
-#     timezone_str = tf.timezone_at(lat=lat, lng=lon)
-
-#     if timezone_str:
+#     tz_name = tf.timezone_at(lat=lat, lng=lon)
+#     if tz_name:
 #         try:
-#             tz = pytz.timezone(timezone_str)
-#             local_dt = datetime.now(tz)
-#             print(f"[✔] Detected Timezone: {timezone_str}")
-#             return local_dt
-#         except Exception as e:
-#             print(f"[⚠] Error loading timezone: {e}")
-#     else:
-#         print("[⚠] Timezone not found. Falling back to UTC.")
-
-#     # Fallback
-#     return datetime.utcnow().replace(tzinfo=pytz.utc)
+#             return pytz.timezone(tz_name)
+#         except Exception:
+#             return pytz.timezone("Asia/Karachi")  # fallback to Pakistan
+#     return pytz.timezone("Asia/Karachi")
 
+# # ---------------------------
 # # Session setup
+# # ---------------------------
 # if "history" not in st.session_state:
 #     st.session_state.history = []
+# if "tz" not in st.session_state:
+#     st.session_state.tz = pytz.timezone("Asia/Karachi")  # default Lahore
 
-# # Page setup
-# st.set_page_config(page_title="AI Disaster Assistant", layout="wide")
+# # ---------------------------
+# # CSS Styling
+# # ---------------------------
 # st.markdown("""
 #     <style>
 #     body { background-color: #0d1117; color: white; }
@@ -67,16 +75,32 @@
 #     .chat-bubble { background-color: #f0f0f0; color: #000; padding: 0.5rem; margin-bottom: 0.5rem; border-radius: 10px; }
 #     .answer-box { background-color: #d6f5f5; padding: 1rem; border-radius: 12px; color: #000; }
 #     .contact-box { background-color: #ffeecc; padding: 1rem; border-radius: 10px; color: #000; }
+#     [data-testid="stSidebar"] { background-color: #161b22; }
+#     .sidebar .sidebar-content { background-color: #161b22; }
 #     </style>
 # """, unsafe_allow_html=True)
 
+# # ---------------------------
+# # Sidebar Navigation
+# # ---------------------------
+# st.sidebar.markdown("# 🌍 AI Disaster Response Assistant")
+# page = st.sidebar.radio("Navigation", ["Chat Assistant", "Image Analysis"])
+
+# if page == "Image Analysis":
+#     st.switch_page("pages/image_analysis.py")
+
+# # ---------------------------
+# # Main Page (Chat Assistant)
+# # ---------------------------
 # st.markdown("""
 #     <h1 style='text-align: center; background-color: #008080; color: white; padding: 1rem; border-radius: 12px;'>
-#     🌍 AI Disaster Response Assistant
+#     💬 AI Disaster Chat Assistant
 #     </h1>
 # """, unsafe_allow_html=True)
 
-# # 🌐 Language Selection
+# # ---------------------------
+# # Language Selection
+# # ---------------------------
 # language_options = {
 #     "English": "en",
 #     "Urdu": "ur",
@@ -88,36 +112,13 @@
 # preferred_language = st.selectbox("🌐 Select your preferred language for responses:", list(language_options.keys()))
 # target_lang_code = language_options[preferred_language]
 
-# # 📍 Get Location
-# # location_data = geocoder.ip('me')
-# # user_location = "Unknown"
-# # city = ""
-# # country = ""
-# # local_time = datetime.utcnow()
-
-# # if location_data and location_data.ok:
-# #     lat = location_data.latlng[0]
-# #     lon = location_data.latlng[1]
-# #     st.success(f"📍 GPS Location: {lat:.4f}, {lon:.4f}")
-# #     try:
-# #         geolocator = Nominatim(user_agent="geoapiExercises")
-# #         location = geolocator.reverse((lat, lon), language='en')
-# #         city = location.raw['address'].get('city', '') or location.raw['address'].get('town', '')
-# #         country = location.raw['address'].get('country', '')
-# #         user_location = f"{city}, {country}"
-# #         st.markdown(f"📌 You are in *{user_location}*")
-
-# #         local_time = get_local_time(lat, lon)
-
-# #     except Exception as e:
-# #         st.warning("🌐 Could not resolve full address or local time from GPS.")
-
+# # ---------------------------
+# # Get Location & Timezone
+# # ---------------------------
 # location_data = get_geolocation()
-
 # user_location = "Unknown"
 # city = ""
 # country = ""
-# local_time = datetime.now()
 
 # if location_data and location_data.get("coords"):
 #     lat = location_data["coords"]["latitude"]
@@ -130,32 +131,63 @@
 #         country = location.raw['address'].get('country', '')
 #         user_location = f"{city}, {country}"
 #         st.markdown(f"📌 You are in *{user_location}*")
+#         st.session_state.tz = get_timezone(lat, lon)
+#     except Exception:
+#         st.warning("🌐 Could not resolve full address or timezone from GPS.")
 
-#         local_time = get_local_time(lat, lon)
-
-#     except Exception as e:
-#         st.warning("🌐 Could not resolve full address or local time from GPS.")
-
-
-# # 🚨 Display rescue numbers
+# # ---------------------------
+# # Display rescue numbers
+# # ---------------------------
 # if country in rescue_contacts:
 #     st.markdown("### 🚨 Local Emergency Numbers")
 #     contacts = rescue_contacts[country]
 #     for service, number in contacts.items():
 #         st.markdown(f"<div class='contact-box'>📞 <strong>{service}</strong>: {number}</div>", unsafe_allow_html=True)
 
+# # ---------------------------
+# # Chat UI
+# # ---------------------------
 # col1, col2 = st.columns([2, 1])
 
 # with col1:
 #     st.markdown("### 🧠 Ask your question (any language):")
-#     user_input = st.text_area(" ", placeholder="e.g. زلزلے کے دوران مجھ کیا کرنا چایے؟ or What to do in a flood?", height=120)
+#     user_input = st.text_area(" ", placeholder="e.g. زلزلے کے دوران مجھے کیا کرنا چاہیے؟ or What to do in a flood?", height=120)
 
 #     if st.button("🔍 Get Help"):
-#         if user_input.strip() != "":
+#         if user_input.strip():
 #             try:
 #                 translated_input = GoogleTranslator(source='auto', target='en').translate(user_input)
-#                 location_context = f"The user is currently in {user_location}. Provide disaster-specific advice and include local rescue contact details if relevant. "
-#                 prompt = f"{location_context}Answer this disaster-related question: {translated_input}"
+#                 # location_context = f"The user is currently in {user_location}. Provide disaster-specific advice and include local rescue contact details if relevant. "
+#                 # Fetch country-specific rescue numbers
+# #                 rescue_info = ""
+# #                 if country in rescue_contacts:
+# #                     rescue_info = "Local emergency numbers:\n" + "\n".join([f"{k}: {v}" for k, v in rescue_contacts[country].items()])
+
+# #                 location_context = f"The user is currently in {user_location}. {rescue_info} Provide disaster-specific advice relevant to this location. If possible, suggest actions based on this specific region."
+# # #     rescue_info = "Local emergency numbers:\n" + "\n".join([f"{k}: {v}" for k, v in rescue_contacts[country].items()])
+
+# # # location_context = f"The user is currently in {user_location}. {rescue_info} Provide disaster-specific advice relevant to this location. If possible, suggest actions based on this specific region."
+
+                
+# #                 prompt = f"{location_context}Answer this disaster-related question: {translated_input}"
+#                 rescue_info = ""
+#                 if country in rescue_contacts:
+#                     contacts = rescue_contacts[country]
+#                     rescue_info = "Local Emergency Numbers:\n" + "\n".join([f"{service}: {number}" for service, number in contacts.items()])
+
+                    
+#                 else:
+#                     rescue_info = "No specific emergency contacts found for this region."
+
+# # Create a location-aware prompt for the LLM
+#                 location_context = (
+#                     f"The user is currently in {user_location}. "
+#                     f"{rescue_info}\n"
+#                     "Provide safety measures specifically for this location in the context of the disaster. "
+#                     "DO NOT mention US-based emergency numbers unless the user's location is in the US."
+# )
+#                 prompt = f"{location_context} Answer this disaster-related question: {translated_input}"
+
 
 #                 headers = {
 #                     "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -173,7 +205,9 @@
 #                     translated_answer = GoogleTranslator(source='en', target=target_lang_code).translate(answer_en)
 #                     st.markdown(f"<div class='answer-box'>{translated_answer}</div>", unsafe_allow_html=True)
 
-#                     timestamp = local_time.strftime("%I:%M %p")
+#                     # Get current local time with timezone
+#                     now = datetime.now(st.session_state.tz)
+#                     timestamp = now.strftime("%I:%M %p")
 #                     st.session_state.history.append({"time": timestamp, "question": user_input})
 #                 else:
 #                     st.error("❌ Error: " + response.text)
@@ -198,231 +232,108 @@
 
 
 import streamlit as st
-import requests
-from datetime import datetime
-from deep_translator import GoogleTranslator
-from geopy.geocoders import Nominatim
+import openai
+from googletrans import Translator
+from streamlit_js_eval import get_geolocation
 from timezonefinder import TimezoneFinder
+from datetime import datetime
 import pytz
-from streamlit_js_eval import get_geolocation
+from geopy.geocoders import Nominatim
+
+st.set_page_config(page_title="AI Disaster Assistant 🌍", layout="centered")
+
+# Session state setup
+if 'tz' not in st.session_state:
+    st.session_state.tz = None
+
+st.title("🌐 AI Disaster Safety Assistant")
+st.markdown("Your emergency translator and safety guide, wherever you are.")
+
+# Get user's geolocation
+loc = get_geolocation()
+
+if loc:
+    lat = loc['coords']['latitude']
+    lon = loc['coords']['longitude']
+    st.success(f"✅ GPS Coordinates detected: Latitude {lat}, Longitude {lon}")
 
-# ---------------------------
-# API Key
-# ---------------------------
-GROQ_API_KEY = "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"
-
-# ---------------------------
-# Page Config
-# ---------------------------
-st.set_page_config(page_title="AI Disaster Assistant", layout="wide", initial_sidebar_state="expanded")
-
-# ---------------------------
-# Rescue numbers dictionary
-# ---------------------------
-rescue_contacts = {
-    "Pakistan": {
-        "Ambulance": "115",
-        "Fire Brigade": "16",
-        "Rescue Service": "1122",
-        "Police": "15"
-    },
-    "India": {
-        "Ambulance": "102",
-        "Fire Brigade": "101",
-        "Disaster Helpline": "108",
-        "Police": "100"
-    },
-    "USA": {
-        "Emergency Services": "911"
-    },
-    "United Kingdom": {
-        "Emergency Services": "999"
-    }
-}
-
-# ---------------------------
-# Timezone Helper
-# ---------------------------
-def get_timezone(lat, lon):
-    tf = TimezoneFinder()
-    tz_name = tf.timezone_at(lat=lat, lng=lon)
-    if tz_name:
-        try:
-            return pytz.timezone(tz_name)
-        except Exception:
-            return pytz.timezone("Asia/Karachi")  # fallback to Pakistan
-    return pytz.timezone("Asia/Karachi")
-
-# ---------------------------
-# Session setup
-# ---------------------------
-if "history" not in st.session_state:
-    st.session_state.history = []
-if "tz" not in st.session_state:
-    st.session_state.tz = pytz.timezone("Asia/Karachi")  # default Lahore
-
-# ---------------------------
-# CSS Styling
-# ---------------------------
-st.markdown("""
-    <style>
-    body { background-color: #0d1117; color: white; }
-    h1, h2, h3, h4, h5 { color: white; }
-    input, textarea, .stTextInput>div>div>input { background-color: #161b22 !important; color: white !important; }
-    .chat-bubble { background-color: #f0f0f0; color: #000; padding: 0.5rem; margin-bottom: 0.5rem; border-radius: 10px; }
-    .answer-box { background-color: #d6f5f5; padding: 1rem; border-radius: 12px; color: #000; }
-    .contact-box { background-color: #ffeecc; padding: 1rem; border-radius: 10px; color: #000; }
-    [data-testid="stSidebar"] { background-color: #161b22; }
-    .sidebar .sidebar-content { background-color: #161b22; }
-    </style>
-""", unsafe_allow_html=True)
-
-# ---------------------------
-# Sidebar Navigation
-# ---------------------------
-st.sidebar.markdown("# 🌍 AI Disaster Response Assistant")
-page = st.sidebar.radio("Navigation", ["Chat Assistant", "Image Analysis"])
-
-if page == "Image Analysis":
-    st.switch_page("pages/image_analysis.py")
-
-# ---------------------------
-# Main Page (Chat Assistant)
-# ---------------------------
-st.markdown("""
-    <h1 style='text-align: center; background-color: #008080; color: white; padding: 1rem; border-radius: 12px;'>
-    💬 AI Disaster Chat Assistant
-    </h1>
-""", unsafe_allow_html=True)
-
-# ---------------------------
-# Language Selection
-# ---------------------------
-language_options = {
-    "English": "en",
-    "Urdu": "ur",
-    "Hindi": "hi",
-    "Japanese": "ja",
-    "Turkish": "tr",
-    "Italian": "it"
-}
-preferred_language = st.selectbox("🌐 Select your preferred language for responses:", list(language_options.keys()))
-target_lang_code = language_options[preferred_language]
-
-# ---------------------------
-# Get Location & Timezone
-# ---------------------------
-location_data = get_geolocation()
-user_location = "Unknown"
-city = ""
-country = ""
-
-if location_data and location_data.get("coords"):
-    lat = location_data["coords"]["latitude"]
-    lon = location_data["coords"]["longitude"]
-    st.success(f"📍 GPS Location: {lat:.4f}, {lon:.4f}")
     try:
+        # Reverse geocoding
         geolocator = Nominatim(user_agent="geoapiExercises")
         location = geolocator.reverse((lat, lon), language='en')
-        city = location.raw['address'].get('city', '') or location.raw['address'].get('town', '')
-        country = location.raw['address'].get('country', '')
-        user_location = f"{city}, {country}"
-        st.markdown(f"📌 You are in *{user_location}*")
-        st.session_state.tz = get_timezone(lat, lon)
-    except Exception:
-        st.warning("🌐 Could not resolve full address or timezone from GPS.")
-
-# ---------------------------
-# Display rescue numbers
-# ---------------------------
-if country in rescue_contacts:
-    st.markdown("### 🚨 Local Emergency Numbers")
-    contacts = rescue_contacts[country]
-    for service, number in contacts.items():
-        st.markdown(f"<div class='contact-box'>📞 <strong>{service}</strong>: {number}</div>", unsafe_allow_html=True)
-
-# ---------------------------
-# Chat UI
-# ---------------------------
-col1, col2 = st.columns([2, 1])
-
-with col1:
-    st.markdown("### 🧠 Ask your question (any language):")
-    user_input = st.text_area(" ", placeholder="e.g. زلزلے کے دوران مجھے کیا کرنا چاہیے؟ or What to do in a flood?", height=120)
-
-    if st.button("🔍 Get Help"):
-        if user_input.strip():
-            try:
-                translated_input = GoogleTranslator(source='auto', target='en').translate(user_input)
-                # location_context = f"The user is currently in {user_location}. Provide disaster-specific advice and include local rescue contact details if relevant. "
-                # Fetch country-specific rescue numbers
-#                 rescue_info = ""
-#                 if country in rescue_contacts:
-#                     rescue_info = "Local emergency numbers:\n" + "\n".join([f"{k}: {v}" for k, v in rescue_contacts[country].items()])
+        if location:
+            address = location.raw.get('address', {})
+            city = (address.get('city') or address.get('town') or 
+                    address.get('village') or address.get('municipality') or '')
+            country = address.get('country', '')
+
+            if city:
+                user_location = f"{city}, {country}"
+            elif country:
+                user_location = f"{country}"
+            else:
+                user_location = "Unknown Location"
+
+            st.markdown(f"📌 You are in **{user_location}**")
+
+            # Timezone calculation
+            tf = TimezoneFinder()
+            timezone_str = tf.timezone_at(lng=lon, lat=lat)
+            if timezone_str:
+                tz = pytz.timezone(timezone_str)
+                local_time = datetime.now(tz).strftime('%Y-%m-%d %H:%M:%S')
+                st.session_state.tz = timezone_str
+                st.markdown(f"🕒 Local time: `{local_time}`")
+            else:
+                st.warning("⚠️ Timezone could not be determined.")
+        else:
+            st.warning("⚠️ Could not find location details for the given coordinates.")
+    except Exception as e:
+        st.warning(f"🌐 Error during location lookup: {e}")
+else:
+    st.warning("📍 Since I couldn't determine your exact location, translation and timezone might be less accurate.")
+
+# Language selector
+language = st.selectbox(
+    "🌍 Select your language",
+    ("English", "Urdu", "Spanish", "French", "Arabic", "Chinese")
+)
 
-#                 location_context = f"The user is currently in {user_location}. {rescue_info} Provide disaster-specific advice relevant to this location. If possible, suggest actions based on this specific region."
-# #     rescue_info = "Local emergency numbers:\n" + "\n".join([f"{k}: {v}" for k, v in rescue_contacts[country].items()])
+# Translator setup
+translator = Translator()
 
-# # location_context = f"The user is currently in {user_location}. {rescue_info} Provide disaster-specific advice relevant to this location. If possible, suggest actions based on this specific region."
+# Input prompt
+user_input = st.text_area("💬 Describe your emergency or ask for help:", "")
 
-                
-#                 prompt = f"{location_context}Answer this disaster-related question: {translated_input}"
-                rescue_info = ""
-                if country in rescue_contacts:
-                    contacts = rescue_contacts[country]
-                    rescue_info = "Local Emergency Numbers:\n" + "\n".join([f"{service}: {number}" for service, number in contacts.items()])
+if st.button("🚨 Get Safety Instructions"):
+    if user_input:
+        prompt = f"Give safety tips and emergency measures based on this situation: {user_input}. Be concise and helpful."
 
-                    
-                else:
-                    rescue_info = "No specific emergency contacts found for this region."
-
-# Create a location-aware prompt for the LLM
-                location_context = (
-                    f"The user is currently in {user_location}. "
-                    f"{rescue_info}\n"
-                    "Provide safety measures specifically for this location in the context of the disaster. "
-                    "DO NOT mention US-based emergency numbers unless the user's location is in the US."
-)
-                prompt = f"{location_context} Answer this disaster-related question: {translated_input}"
-
-
-                headers = {
-                    "Authorization": f"Bearer {GROQ_API_KEY}",
-                    "Content-Type": "application/json"
-                }
-                data = {
-                    "model": "llama3-70b-8192",
-                    "messages": [{"role": "user", "content": prompt}]
-                }
-
-                response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=data)
-
-                if response.status_code == 200:
-                    answer_en = response.json()['choices'][0]['message']['content']
-                    translated_answer = GoogleTranslator(source='en', target=target_lang_code).translate(answer_en)
-                    st.markdown(f"<div class='answer-box'>{translated_answer}</div>", unsafe_allow_html=True)
-
-                    # Get current local time with timezone
-                    now = datetime.now(st.session_state.tz)
-                    timestamp = now.strftime("%I:%M %p")
-                    st.session_state.history.append({"time": timestamp, "question": user_input})
-                else:
-                    st.error("❌ Error: " + response.text)
-            except Exception as e:
-                st.error(f"⚠ Exception occurred: {e}")
-
-with col2:
-    st.markdown("### 📜 Chat History")
-    if st.session_state.history:
-        for entry in reversed(st.session_state.history):
-            st.markdown(f"""
-                <div class='chat-bubble'>
-                    🕒 {entry['time']}<br><strong>{entry['question']}</strong>
-                </div>
-            """, unsafe_allow_html=True)
-    else:
-        st.info("No questions asked yet.")
+        openai.api_key = st.secrets.get("GROQ_API_KEY", "your-api-key")
 
-    if st.button("🧹 Clear History"):
-        st.session_state.history = []
-        st.success("Chat history cleared.")
+        try:
+            from openai import OpenAI
+            client = OpenAI()
+            completion = client.chat.completions.create(
+                model="llama3-70b-8192",
+                messages=[
+                    {"role": "system", "content": "You are a helpful assistant providing safety tips."},
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0.7
+            )
+            reply = completion.choices[0].message.content
+
+            # Translate if needed
+            if language != "English":
+                translated = translator.translate(reply, dest=language.lower())
+                st.markdown("### 🛡️ Safety Tips (Translated):")
+                st.write(translated.text)
+            else:
+                st.markdown("### 🛡️ Safety Tips:")
+                st.write(reply)
+
+        except Exception as e:
+            st.error(f"❌ Error generating response: {e}")
+    else:
+        st.warning("Please describe your emergency situation first.")