Update app.py

app.py

@@ -1,64 +1,45 @@
 import gradio as gr
 from mistralai import Mistral
-import os
 
-# -----------------------------
-# 1. Initialize Mistral client
-# -----------------------------
-MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY")
-
-if not MISTRAL_API_KEY:
-    raise ValueError("Please set your MISTRAL_API_KEY in Hugging Face Space secrets.")
-
-client = Mistral(api_key=MISTRAL_API_KEY)
-
-# -----------------------------
-# 2. Chatbot function
-# -----------------------------
-def chat_with_mistral(history, user_input):
-    """
-    Send conversation to Mistral API and return the model's response.
-    """
-    # Prepare messages from chat history
-    messages = [{"role": "system", "content": "You are a helpful AI assistant."}]
-    for msg in history:
-        messages.append({"role": "user", "content": msg[0]})
-        messages.append({"role": "assistant", "content": msg[1]})
-    messages.append({"role": "user", "content": user_input})
+# 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="YOUR_MISTRAL_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-large-latest",  # Mistral's latest large language model
-            messages=messages,
+            model="mistral-medium",
+            messages=messages
         )
-        bot_reply = response.choices[0].message["content"]
-    except Exception as e:
-        bot_reply = f"⚠️ Error: {str(e)}"
 
-    return bot_reply
-
-# -----------------------------
-# 3. Gradio Interface
-# -----------------------------
-def chat_interface(user_input, history):
-    reply = chat_with_mistral(history, user_input)
-    history.append((user_input, reply))
-    return history, history
-
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
-    gr.Markdown("# 💬 Mistral-Powered Chatbot")
-    gr.Markdown("Chat with an AI assistant powered by **Mistral** and **Gradio**.")
-
-    chatbot = gr.Chatbot(label="Mistral Chatbot", height=500)
-    user_input = gr.Textbox(label="Your message", placeholder="Type your question here...")
-    clear_btn = gr.Button("Clear Chat")
+        # ✅ 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)}"
+        history.append((user_input, error_msg))
+        return history, ""
 
-    state = gr.State([])
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## 💬 Mistral Chatbot — Skin & Acne Specialist")
 
-    user_input.submit(chat_interface, [user_input, state], [chatbot, state])
-    clear_btn.click(lambda: ([], []), None, [chatbot, state])
+    chatbot = gr.Chatbot(height=500)
+    msg = gr.Textbox(label="Ask your question here:")
+    clear = gr.ClearButton([msg, chatbot])
 
-    gr.Markdown("🧠 Model: **mistral-large-latest** | ⚙️ Built with Gradio")
+    msg.submit(chat_with_mistral, [msg, chatbot], [chatbot, msg])
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()