Update app.py

app.py

@@ -1,3 +1,4 @@
+# --- Imports ---
 import streamlit as st
 import PyPDF2
 import requests
@@ -5,10 +6,10 @@ import json
 from sentence_transformers import SentenceTransformer
 import numpy as np
 from sklearn.metrics.pairwise import cosine_similarity
-import os
 from datetime import datetime
+from groq import Groq
 
-# Page configuration
+# --- Page config ---
 st.set_page_config(
     page_title="🚨 First Aid Emergency Assistant",
     page_icon="🚨",
@@ -16,7 +17,7 @@ st.set_page_config(
     initial_sidebar_state="collapsed"
 )
 
-# Custom CSS for Professional ChatGPT-like interface
+# --- Enhanced Professional UI CSS ---
 st.markdown("""
 <style>
     /* Main App Background */
@@ -47,13 +48,6 @@ st.markdown("""
         background-clip: text;
     }
     
-    .main-header p {
-        margin: 0.5rem 0 0 0;
-        font-size: 1.2rem;
-        opacity: 0.8;
-        color: #a0aec0;
-    }
-    
     /* Chat Container */
     .chat-container {
         background: rgba(26, 32, 46, 0.8);
@@ -185,6 +179,26 @@ st.markdown("""
         transform: translateY(0) !important;
     }
     
+    /* Form Submit Button Styling */
+    .stForm > div > div > button {
+        background: linear-gradient(135deg, #4FACFE 0%, #00F2FE 100%) !important;
+        color: white !important;
+        border: none !important;
+        border-radius: 30px !important;
+        padding: 16px 32px !important;
+        font-weight: 600 !important;
+        font-size: 16px !important;
+        transition: all 0.3s ease !important;
+        box-shadow: 0 4px 20px rgba(79, 172, 254, 0.3) !important;
+        width: 100% !important;
+    }
+    
+    .stForm > div > div > button:hover {
+        transform: translateY(-2px) !important;
+        box-shadow: 0 8px 30px rgba(79, 172, 254, 0.5) !important;
+        background: linear-gradient(135deg, #00F2FE 0%, #4FACFE 100%) !important;
+    }
+    
     /* Warning Box */
     .warning-box {
         background: linear-gradient(135deg, rgba(255, 107, 107, 0.1) 0%, rgba(255, 154, 158, 0.1) 100%);
@@ -197,24 +211,25 @@ st.markdown("""
         backdrop-filter: blur(10px);
     }
     
-    /* Sidebar Styling */
-    .sidebar-info {
-        background: rgba(45, 55, 72, 0.8);
-        padding: 1.5rem;
-        border-radius: 15px;
-        margin: 1rem 0;
-        border: 1px solid rgba(79, 172, 254, 0.2);
-        color: #e2e8f0;
+    /* Hide Streamlit default elements */
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
+    header {visibility: hidden;}
+    
+    /* Bot message formatting */
+    .bot-message strong {
+        color: #00F2FE !important;
     }
     
-    /* Loading Spinner Custom */
-    .stSpinner > div {
-        border-top-color: #4FACFE !important;
+    .bot-message h1, .bot-message h2, .bot-message h3 {
+        color: #4FACFE !important;
+        margin-top: 1rem !important;
     }
     
-    /* Typing Indicator */
+    /* Typing indicator */
     .typing-indicator {
         background: rgba(45, 55, 72, 0.9);
+        color: #a0aec0;
         padding: 16px 20px;
         border-radius: 20px 20px 20px 8px;
         margin: 16px auto 16px 0;
@@ -230,349 +245,164 @@ st.markdown("""
         50% { opacity: 1; }
         100% { opacity: 0.6; }
     }
-    
-    /* Custom scrollbar for main app */
-    .main .block-container {
-        padding-top: 2rem;
-    }
-    
-    /* Hide Streamlit default elements */
-    #MainMenu {visibility: hidden;}
-    footer {visibility: hidden;}
-    header {visibility: hidden;}
-    
-    /* Custom markdown styling */
-    .bot-message h1, .bot-message h2, .bot-message h3 {
-        color: #4FACFE !important;
-        margin-top: 1rem !important;
-    }
-    
-    .bot-message ul, .bot-message ol {
-        padding-left: 1.5rem !important;
-    }
-    
-    .bot-message strong {
-        color: #00F2FE !important;
-    }
-    
-    .bot-message code {
-        background: rgba(79, 172, 254, 0.1) !important;
-        padding: 2px 6px !important;
-        border-radius: 4px !important;
-        color: #4FACFE !important;
-    }
 </style>
 """, unsafe_allow_html=True)
 
-# Initialize GROQ API
+# --- GROQ API setup ---
 @st.cache_resource
 def setup_groq():
-    groq_api_key = st.secrets.get("GROQ_API_KEY") or os.getenv("GROQ_API_KEY")
-    if not groq_api_key:
-        st.error("⚠️ GROQ API key not found! Please add it to your Hugging Face secrets.")
+    try:
+        return st.secrets["GROQ_API_KEY"]
+    except:
+        st.error("❌ Missing GROQ_API_KEY in `.streamlit/secrets.toml`")
         st.stop()
-    return groq_api_key
 
-# Load and process PDF
+client = Groq(api_key=setup_groq())
+
+# --- PDF Processing ---
 @st.cache_resource
 def load_pdf():
     try:
-        with open("First-Aid.pdf", "rb") as file:
-            pdf_reader = PyPDF2.PdfReader(file)
-            text = ""
-            for page in pdf_reader.pages:
-                text += page.extract_text() + "\n"
-        return text
-    except FileNotFoundError:
-        st.error("📄 First-Aid.pdf not found! Please upload the PDF file to your space.")
-        st.stop()
-    except Exception as e:
-        st.error(f"❌ Error loading PDF: {str(e)}")
+        with open("First-Aid.pdf", "rb") as f:
+            reader = PyPDF2.PdfReader(f)
+            text = "".join(page.extract_text() + "\n" for page in reader.pages)
+            return text
+    except:
+        st.error("❌ Failed to load 'First-Aid.pdf'. Upload it to the app root.")
         st.stop()
 
-# Setup embeddings and knowledge base
+# --- Knowledge Base Setup ---
 @st.cache_resource
 def setup_knowledge_base():
-    # Load PDF content
-    pdf_text = load_pdf()
-    
-    # Split text into chunks
-    chunks = []
-    sentences = pdf_text.split('\n')
-    current_chunk = ""
-    
-    for sentence in sentences:
+    text = load_pdf()
+    chunks, current_chunk = [], ""
+    for sentence in text.split('\n'):
         if len(current_chunk + sentence) < 1000:
             current_chunk += sentence + "\n"
         else:
-            if current_chunk.strip():
-                chunks.append(current_chunk.strip())
+            if current_chunk.strip(): chunks.append(current_chunk.strip())
             current_chunk = sentence + "\n"
-    
-    if current_chunk.strip():
-        chunks.append(current_chunk.strip())
-    
-    # Load sentence transformer model
+    if current_chunk.strip(): chunks.append(current_chunk.strip())
     model = SentenceTransformer('all-MiniLM-L6-v2')
-    
-    # Create embeddings for chunks
-    chunk_embeddings = model.encode(chunks)
-    
-    return chunks, chunk_embeddings, model
+    embeddings = model.encode(chunks)
+    return chunks, embeddings, model
 
-def find_relevant_context(query, chunks, chunk_embeddings, model, top_k=3):
-    """Find most relevant chunks for the query"""
+# --- Context Matching ---
+def find_relevant_context(query, chunks, embeddings, model, top_k=3):
     query_embedding = model.encode([query])
-    similarities = cosine_similarity(query_embedding, chunk_embeddings)[0]
+    similarities = cosine_similarity(query_embedding, embeddings)[0]
     top_indices = np.argsort(similarities)[-top_k:][::-1]
-    
-    relevant_chunks = [chunks[i] for i in top_indices]
-    return "\n\n".join(relevant_chunks)
+    return "\n\n".join([chunks[i] for i in top_indices])
 
+# --- Query GROQ ---
 def query_groq(prompt, api_key):
-    """Query GROQ API"""
     url = "https://api.groq.com/openai/v1/chat/completions"
-    
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
     data = {
-        "model": "mixtral-8x7b-32768",
+        "model": "llama3-70b-8192",
         "messages": [
             {
                 "role": "system",
-                "content": """You are a First Aid Emergency Assistant. You provide clear, step-by-step first aid guidance based on the provided medical manual context. 
-
-IMPORTANT RULES:
-1. Only answer questions related to first aid, medical emergencies, and health safety
-2. If asked about non-medical topics, politely redirect to first aid topics
-3. For serious emergencies, always remind users to call emergency services first
-4. Provide clear, numbered steps when giving instructions
-5. Keep responses focused and practical
-
-If the question is not related to first aid or medical emergencies, respond with: "🚨 I'm specialized in First Aid emergencies only! Please ask me about medical emergencies, CPR, wounds, burns, fractures, or other first aid topics."
-"""
+                "content": """You are a First Aid Emergency Assistant. Answer only first aid-related queries. Be clear, concise, and step-by-step. For serious emergencies, advise calling 911. If asked off-topic, respond with: "🚨 I'm specialized in First Aid emergencies only!"."""
             },
-            {
-                "role": "user",
-                "content": prompt
-            }
+            {"role": "user", "content": prompt}
         ],
         "temperature": 0.3,
         "max_tokens": 1000
     }
-    
     try:
-        response = requests.post(url, headers=headers, json=data, timeout=30)
-        response.raise_for_status()
-        return response.json()["choices"][0]["message"]["content"]
-    except requests.exceptions.RequestException as e:
-        return f"❌ Error connecting to GROQ API: {str(e)}"
+        r = requests.post(url, headers=headers, json=data, timeout=30)
+        r.raise_for_status()
+        return r.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        return f"❌ Error processing response: {str(e)}"
+        return f"❌ GROQ API Error: {str(e)}"
 
-# Initialize session state
+# --- Session State ---
 if "messages" not in st.session_state:
-    st.session_state.messages = [
-        {
-            "role": "assistant", 
-            "content": "🚨 **Hello! I'm your First Aid Emergency Assistant.**\n\nI can help you with:\n• CPR procedures\n• Bleeding control\n• Burns treatment\n• Choking response\n• Fracture management\n• Poisoning emergencies\n• And much more!\n\n💡 **Ask me anything about first aid emergencies!**"
-        }
-    ]
+    st.session_state.messages = [{
+        "role": "assistant",
+        "content": "🚨 **Hello! I'm your First Aid Emergency Assistant.**\n\nAsk me about CPR, bleeding, choking, burns, or any first aid emergency."
+    }]
 
 if "knowledge_base" not in st.session_state:
-    with st.spinner("🔄 Loading First Aid knowledge base..."):
+    with st.spinner("Loading knowledge base..."):
         chunks, embeddings, model = setup_knowledge_base()
         st.session_state.knowledge_base = {
-            "chunks": chunks,
-            "embeddings": embeddings,
-            "model": model
+            "chunks": chunks, "embeddings": embeddings, "model": model
         }
 
-if "groq_api_key" not in st.session_state:
-    st.session_state.groq_api_key = setup_groq()
+# --- Chat UI ---
+st.markdown('<div class="main-header"><h1>🚨 First Aid Emergency Assistant</h1><p style="margin: 0.5rem 0 0 0; font-size: 1.2rem; opacity: 0.8; color: #a0aec0;">Your AI-powered emergency response guide</p></div>', unsafe_allow_html=True)
 
-# Header
-st.markdown("""
-<div class="main-header">
-    <h1>🚨 First Aid Emergency Assistant</h1>
-    <p style="margin: 0; font-size: 18px; opacity: 0.9;">Your AI-powered emergency response guide</p>
-</div>
-""", unsafe_allow_html=True)
-
-# Warning disclaimer
+# Disclaimer
 st.markdown("""
 <div class="warning-box">
-    <strong>⚠️ IMPORTANT MEDICAL DISCLAIMER:</strong><br>
-    This chatbot provides general first aid guidance only. In real emergencies, always call emergency services immediately. 
-    This tool is not a substitute for professional medical advice, diagnosis, or treatment.
+<strong>⚠️ MEDICAL DISCLAIMER:</strong><br>
+This tool is for informational purposes only. Always call emergency services in real situations.
 </div>
 """, unsafe_allow_html=True)
 
-# Chat container
-chat_container = st.container()
-
-with chat_container:
+# Chat display
+with st.container():
     st.markdown('<div class="chat-container">', unsafe_allow_html=True)
-    
-    # Display chat messages
-    for message in st.session_state.messages:
-        if message["role"] == "user":
-            st.markdown(f"""
-            <div class="user-message">
-                {message["content"]}
-            </div>
-            <div style="clear: both;"></div>
-            """, unsafe_allow_html=True)
+    for msg in st.session_state.messages:
+        if msg["role"] == "user":
+            st.markdown(f'<div class="user-message">{msg["content"]}</div><div style="clear: both;"></div>', unsafe_allow_html=True)
         else:
-            st.markdown(f"""
-            <div class="bot-message">
-                <strong>🤖 First Aid Assistant</strong><br>
-                {message["content"]}
-            </div>
-            <div style="clear: both;"></div>
-            """, unsafe_allow_html=True)
-    
+            st.markdown(f'<div class="bot-message"><strong>🤖 First Aid Assistant</strong><br>{msg["content"]}</div><div style="clear: both;"></div>', unsafe_allow_html=True)
     st.markdown('</div>', unsafe_allow_html=True)
 
-# Input section with Enter key support
+# Enhanced Input UI with Enter key support and auto-clear
 st.markdown('<div class="input-container">', unsafe_allow_html=True)
 
-# Create a form to handle Enter key submission
+# Use form for Enter key support and auto-clear
 with st.form(key="chat_form", clear_on_submit=True):
     col1, col2 = st.columns([4, 1])
-    
     with col1:
-        user_input = st.text_input(
-            "",
-            placeholder="Ask me about first aid emergencies... (Press Enter or click Send)",
-            key="user_input_form",
-            label_visibility="collapsed"
-        )
-    
+        user_input = st.text_input("", placeholder="Ask about a first aid emergency... (Press Enter or click Send)", key="user_input_form", label_visibility="collapsed")
     with col2:
-        send_button = st.form_submit_button("Send 🚀")
+        send = st.form_submit_button("Send 🚀")
 
 st.markdown('</div>', unsafe_allow_html=True)
 
-# Process user input (from either Enter key or Send button)
-if send_button and user_input and user_input.strip():
-    # Add user message to chat
+# Process Input (works with both Enter key and button click)
+if send and user_input and user_input.strip():
+    # Add user message
     st.session_state.messages.append({"role": "user", "content": user_input})
     
-    # Show typing indicator
-    with st.container():
-        st.markdown('<div class="chat-container">', unsafe_allow_html=True)
-        # Display all previous messages
-        for message in st.session_state.messages:
-            if message["role"] == "user":
-                st.markdown(f"""
-                <div class="user-message">
-                    {message["content"]}
-                </div>
-                <div style="clear: both;"></div>
-                """, unsafe_allow_html=True)
-            else:
-                st.markdown(f"""
-                <div class="bot-message">
-                    <strong>🤖 First Aid Assistant</strong><br>
-                    {message["content"]}
-                </div>
-                <div style="clear: both;"></div>
-                """, unsafe_allow_html=True)
-        
-        # Show typing indicator
+    # Show typing indicator temporarily
+    with st.empty():
         st.markdown('''
-        <div class="typing-indicator">
-            <strong>🤖 First Aid Assistant</strong><br>
-            <em>Typing...</em>
+        <div class="chat-container">
+            <div class="typing-indicator">
+                <strong>🤖 First Aid Assistant</strong><br>
+                <em>Typing...</em>
+            </div>
+            <div style="clear: both;"></div>
         </div>
-        <div style="clear: both;"></div>
         ''', unsafe_allow_html=True)
-        st.markdown('</div>', unsafe_allow_html=True)
     
-    # Get bot response
+    # Get response
     try:
-        # Find relevant context from PDF
         kb = st.session_state.knowledge_base
-        context = find_relevant_context(
-            user_input, 
-            kb["chunks"], 
-            kb["embeddings"], 
-            kb["model"]
-        )
-        
-        # Create enhanced prompt with context
-        enhanced_prompt = f"""
-Based on the following first aid manual content, answer this question: {user_input}
+        context = find_relevant_context(user_input, kb["chunks"], kb["embeddings"], kb["model"])
+        full_prompt = f"""
+Based on this manual content, answer the question: {user_input}
 
-Context from First Aid Manual:
+Context:
 {context}
-
-Please provide a clear, helpful response based on this information. If this is a serious emergency, remind the user to call emergency services first.
 """
-        
-        # Query GROQ API
-        response = query_groq(enhanced_prompt, st.session_state.groq_api_key)
-        
-        # Enhance response with emergency reminder for serious cases
-        serious_keywords = ['heart attack', 'stroke', 'unconscious', 'not breathing', 'severe bleeding', 'poisoning', 'choking']
-        if any(keyword in user_input.lower() for keyword in serious_keywords):
-            response = f"🚨 **CALL EMERGENCY SERVICES IMMEDIATELY!**\n\n{response}"
-            
-    except Exception as e:
-        response = f"❌ Sorry, I encountered an error: {str(e)}. Please try asking your question differently."
+        response = query_groq(full_prompt, setup_groq())
 
-    # Add bot response to chat
-    st.session_state.messages.append({"role": "assistant", "content": response})
-    
-    # Rerun to show new message without typing indicator
-    st.rerun()
+        # Add emergency warning for serious cases
+        if any(x in user_input.lower() for x in ["heart attack", "stroke", "not breathing", "unconscious", "severe bleeding"]):
+            response = f"🚨 **CALL EMERGENCY SERVICES IMMEDIATELY!**\n\n{response}"
 
-# Sidebar with helpful information
-with st.sidebar:
-    st.markdown("## 📋 Quick Emergency Numbers")
-    st.markdown("""
-    <div class="sidebar-info">
-    <strong>🚨 Emergency Services:</strong><br>
-    • General Emergency: 911<br>
-    • Poison Control: 1-800-222-1222<br>
-    • Mental Health Crisis: 988
-    </div>
-    """, unsafe_allow_html=True)
-    
-    st.markdown("## 🎯 What I Can Help With")
-    st.markdown("""
-    <div class="sidebar-info">
-    • CPR and rescue breathing<br>
-    • Wound care and bleeding<br>
-    • Burns and scalds<br>
-    • Fractures and sprains<br>
-    • Choking procedures<br>
-    • Poisoning emergencies<br>
-    • Heart attack signs<br>
-    • Snake and animal bites<br>
-    • Drowning response<br>
-    • And much more!
-    </div>
-    """, unsafe_allow_html=True)
+        # Add bot response
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        
+    except Exception as e:
+        st.session_state.messages.append({"role": "assistant", "content": f"❌ Sorry, I encountered an error: {str(e)}. Please try again."})
     
-    st.markdown("## ℹ️ How to Use")
-    st.markdown("""
-    <div class="sidebar-info">
-    1. Type your first aid question<br>
-    2. Get instant step-by-step guidance<br>
-    3. Follow instructions carefully<br>
-    4. Seek professional help for serious emergencies
-    </div>
-    """, unsafe_allow_html=True)
-
-# Footer
-st.markdown("---")
-st.markdown("""
-<div style="text-align: center; opacity: 0.7; padding: 1rem;">
-    🤖 First Aid Emergency Assistant | Powered by GROQ AI | Always consult medical professionals for serious emergencies
-</div>
-""", unsafe_allow_html=True)
+    # Rerun to show updated chat
+    st.rerun()