Update app.py

app.py

@@ -7,6 +7,9 @@ from datetime import datetime
 from groq import Groq
 import pdfplumber
 import plotly.graph_objects as go
+import folium
+from streamlit_folium import st_folium
+from streamlit_geolocation import streamlit_geolocation
 
 # --- 1. CORE SYSTEM CONFIG ---
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
@@ -38,7 +41,6 @@ st.markdown("""
         background-color: #1e3a8a !important;
         color: white !important;
         border: 1px solid #3b82f6 !important;
-        box-shadow: none !important;
     }
 
     /* User Message: Beautiful Blue Gradient */
@@ -77,6 +79,11 @@ def send_call_request(caller, receiver, room):
     new_req = pd.DataFrame([{"Caller": caller, "Receiver": receiver, "RoomID": room, "Status": "Active"}])
     pd.concat([df, new_req]).to_csv(CALL_SIGNAL_DB, index=False)
 
+def clear_call_request(receiver):
+    df = load_db(CALL_SIGNAL_DB, ["Caller", "Receiver", "RoomID", "Status"])
+    df = df[df['Receiver'] != receiver]
+    df.to_csv(CALL_SIGNAL_DB, index=False)
+
 # Session Initialization
 if "logged_in" not in st.session_state: st.session_state.logged_in = False
 if "msgs" not in st.session_state: st.session_state.msgs = []
@@ -111,29 +118,28 @@ if not st.session_state.logged_in:
                     st.warning("User already exists.")
                 else:
                     pd.concat([df, pd.DataFrame([{"username": nu, "password": hash_pass(np), "role": nr}])]).to_csv(USER_DB, index=False)
-                    st.success("Account Registered! You can now log in.") #
+                    st.success("Account Registered! Login now.") #
     st.stop()
 
 # --- 5. SIDEBAR NAVIGATION ---
 with st.sidebar:
     st.markdown(f"### 👤 {st.session_state.username} ({st.session_state.role})")
-    if st.button("🗑️ Clear Chat"): # Reset conversation
+    if st.button("🗑️ Clear Chat"):
         st.session_state.msgs = []
         st.session_state.active_doc = None
         st.rerun()
     if st.button("🚪 Logout"): st.session_state.logged_in = False; st.rerun()
     st.divider()
     if st.session_state.role == "Patient":
-        nav = st.radio("Menu", ["💬 AI Chat", "🧪 Health Tools", "📞 Video Consult", "📜 Call History"])
+        nav = st.radio("Menu", ["💬 AI Chat", "🧪 Health Tools", "📍 Nearby Clinics", "📞 Video Consult", "📜 Call History"])
     else:
         nav = st.radio("Menu", ["🖥️ Consultation Desk", "📜 Call Logs"])
 
 # --- 6. AI CHAT MODULE ---
 if nav == "💬 AI Chat":
     st.markdown("### 💬 Clinical Intelligence Assistant")
-    
     if st.session_state.active_doc:
-        st.info("📂 Medical document analyzed.")
+        st.info("📂 Medical document active. Analysis mode engaged.")
         st.download_button("📄 Download PDF Summary", data=st.session_state.active_doc, file_name="medical_summary.txt") #
     
     for m in st.session_state.msgs:
@@ -161,13 +167,7 @@ if nav == "💬 AI Chat":
 
     if final_q:
         st.session_state.msgs.append({"role": "user", "content": final_q})
-        sys_p = (
-            "You are the IntelliCare Medical Assistant. "
-            "1. Greet the user professionally as a medical assistant. "
-            "2. Strictly answer ONLY medical/health queries. "
-            "3. Refuse essays, code, or non-medical topics. "
-            "4. Prioritize PDF analysis if context is present."
-        )
+        sys_p = "Strict Medical Assistant. Refuse essays/code. Analyze PDFs specifically."
         ans = Groq(api_key=GROQ_API_KEY).chat.completions.create(model="llama-3.3-70b-versatile", messages=[{"role": "system", "content": sys_p}, {"role": "system", "content": f"PDF_CONTEXT: {st.session_state.active_doc}"}] + st.session_state.msgs)
         st.session_state.msgs.append({"role": "assistant", "content": ans.choices[0].message.content})
         st.rerun()
@@ -196,7 +196,19 @@ elif nav == "🧪 Health Tools":
         fig = go.Figure(data=go.Scatter(x=t, y=ecg, line=dict(color='#ef4444')))
         st.plotly_chart(fig, use_container_width=True)
 
-# --- 8. CALLS & HISTORY ---
+# --- 8. MAP FEATURE (NEARBY CLINICS) ---
+elif nav == "📍 Nearby Clinics":
+    st.markdown("### 📍 Specialist Hospital Locator")
+    loc = streamlit_geolocation()
+    if loc.get("latitude"):
+        # Display map based on current geolocation
+        m = folium.Map(location=[loc["latitude"], loc["longitude"]], zoom_start=14)
+        folium.Marker([loc["latitude"], loc["longitude"]], popup="You are here", icon=folium.Icon(color='red')).add_to(m)
+        st_folium(m, width=1000, height=500)
+    else:
+        st.info("Allow location access to find nearby medical facilities.")
+
+# --- 9. CALLS & HISTORY ---
 elif nav == "📞 Video Consult":
     st.markdown("### 📞 Specialist Call Request")
     db = load_db(USER_DB, ["username", "role"])
@@ -217,7 +229,14 @@ elif nav == "🖥️ Consultation Desk":
     my_calls = signals[signals['Receiver'] == st.session_state.username]
     if not my_calls.empty:
         st.info(f"🔔 Incoming request from **{my_calls.iloc[0]['Caller']}**")
-        if st.button("Accept Call"): st.session_state.d_room = my_calls.iloc[0]['RoomID']
+        c1, c2 = st.columns(2)
+        with c1:
+            if st.button("✅ Accept & Join"): st.session_state.d_room = my_calls.iloc[0]['RoomID']
+        with c2:
+            if st.button("❌ Decline Call"): #
+                clear_call_request(st.session_state.username)
+                st.warning("Call declined.")
+                st.rerun()
     else: st.write("Waiting for requests...")
     if "d_room" in st.session_state:
         st.components.v1.html(f'<iframe src="https://meet.jit.si/{st.session_state.d_room}" width="100%" height="600px"></iframe>', height=650)