Update app.py

app.py

@@ -2,21 +2,41 @@ from groq import Groq
 import streamlit as st
 import re
 from datetime import datetime
-# import os
+import os
 from typing import Generator, List, Tuple, Optional
 import logging
-# from dotenv import load_dotenv
+from dotenv import load_dotenv
 
 # --- Load Environment Variables ---
-# load_dotenv()
+load_dotenv()
 
 # --- Logging Configuration ---
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# --- API Key Management ---
+def get_api_key() -> Optional[str]:
+    """Get API key from environment or user input."""
+    api_key = os.getenv("GROQ_API_KEY")
+    
+    if not api_key:
+        st.sidebar.markdown("## 🔑 API Configuration")
+        api_key = st.sidebar.text_input(
+            "Enter your Groq API Key:",
+            type="password",
+            help="Get your API key from https://console.groq.com",
+            key="groq_api_key"
+        )
+        if api_key:
+            st.sidebar.success("API Key set successfully!")
+        else:
+            st.sidebar.warning("Please enter your Groq API Key to continue")
+    
+    return api_key
+
 # --- Constants ---
 MODEL_CONFIG = {
-    "model": "meta-llama/Llama-Guard-4-12B",
+    "model": "meta-llama/llama-4-scout-17b-16e-instruct",
     "temperature": 0.5,
     "max_completion_tokens": 1024,
     "stream": True,
@@ -186,8 +206,17 @@ def main() -> None:
     try:
         setup_page()
         initialize_session_state()
+        
+        # Get API key
+        api_key = get_api_key()
+        if not api_key:
+            st.stop()
+            
+        # Initialize Groq client
+        global client
+        client = Groq(api_key=api_key)
+        
         render_sidebar()
-
         st.title("🧠 DoctorX")
         st.caption("Preliminary health guidance - Always consult healthcare professionals")
 
@@ -206,16 +235,6 @@ def main() -> None:
         logger.error(f"Application error: {str(e)}")
         st.error("An unexpected error occurred. Please refresh the page and try again.")
 
-# --- LLM Initialization ---
-try:
-    # client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-    client = Groq(api_key=GROQ_API_KEY)
-    if not client:
-        raise ValueError("Failed to initialize Groq client. Please check your API key.")
-except Exception as e:
-    logger.error(f"Error initializing Groq client: {str(e)}")
-    st.error("Failed to initialize AI model. Please check your API key.")
-    st.stop()
-
+# Remove the global client initialization
 if __name__ == "__main__":
     main()