Update app.py

app.py

@@ -1,38 +1,38 @@
+import os
 import gradio as gr
 from mistralai import Mistral
 
-# Initialize Mistral client with your API key
-# Set your API key in the Hugging Face Space secret settings
-# (Settings → Variables → Add "MISTRAL_API_KEY")
-client = Mistral(api_key="MISTRAL_API_KEY")
+# Fetch the API key from environment variables (set in Hugging Face secrets)
+api_key = os.getenv("MISTRAL_API_KEY")
+
+if not api_key:
+    raise ValueError("⚠️ Missing MISTRAL_API_KEY. Please set it in the Space settings.")
+
+client = Mistral(api_key=api_key)
 
 def chat_with_mistral(user_input, history):
     """Handles chat with Mistral LLM."""
     try:
-        # Convert chat history to message format
         messages = [{"role": "system", "content": "You are a helpful medical assistant specialized in skin and acne care."}]
         for human, ai in history:
             messages.append({"role": "user", "content": human})
             messages.append({"role": "assistant", "content": ai})
         messages.append({"role": "user", "content": user_input})
 
-        # Query Mistral API
         response = client.chat.complete(
             model="mistral-medium",
             messages=messages
         )
 
-        # ✅ Correct way to extract text
         reply = response.choices[0].message.content
         history.append((user_input, reply))
         return history, ""
     
     except Exception as e:
-        error_msg = f"⚠️ Error: {str(e)}"
+        error_msg = f"⚠️ API error occurred: {str(e)}"
         history.append((user_input, error_msg))
         return history, ""
 
-# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## 💬 Mistral Chatbot — Skin & Acne Specialist")