Update app.py

app.py

@@ -1,22 +1,35 @@
-import streamlit as st
-import requests
-import os
-
-# Hugging Face API details
-API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
-headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}  # read from secret
-
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
-
-# Streamlit UI
-st.title("🚀 Mistral-7B AI Chat App")
-
-user_input = st.text_input("Ask me anything:")
-
-if st.button("Send"):
-    if user_input:
-        output = query({"inputs": user_input})
-        st.write("### 🤖 AI Response:")
-        st.write(output[0]["generated_text"])
+import os
+import requests
+import streamlit as st
+
+# Get token securely
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
+headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+st.set_page_config(page_title="Mistral Chatbot", page_icon="🤖")
+st.title("🚀 Mistral-7B AI Chat App")
+
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+user_input = st.text_input("Ask me anything:")
+
+if st.button("Send"):
+    if user_input:
+        output = query({"inputs": user_input})
+        try:
+            reply = output[0]["generated_text"]
+        except:
+            reply = "⚠️ Something went wrong. Please try again."
+
+        st.session_state.history.append(("You", user_input))
+        st.session_state.history.append(("AI", reply))
+
+for role, text in st.session_state.history:
+    st.markdown(f"**{role}:** {text}")