Update app.py

app.py

@@ -1,20 +1,16 @@
 import streamlit as st
 from transformers import pipeline
-from huggingface_hub import login, whoami
+from huggingface_hub import login
 
-# Load API key securely from Streamlit secrets
-HF_TOKEN = st.secrets["huggingface"]["api_token"]
-
-# Verify token before logging in
+# ✅ Retrieve API token securely
 try:
-    user_info = whoami(HF_TOKEN)
-    st.write(f"✅ Logged in as: {user_info['name']}")
-    login(HF_TOKEN)
-except Exception as e:
-    st.error(f"❌ Invalid Hugging Face Token: {e}")
+    HF_TOKEN = st.secrets["huggingface"]["api_token"]
+    login(HF_TOKEN)  # Log in to Hugging Face Hub
+except Exception:
+    st.error("❌ Missing or invalid Hugging Face API token. Please check secrets.toml.")
     st.stop()
 
-# Load the model
+# ✅ Load the model
 @st.cache_resource
 def load_model():
     return pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", token=HF_TOKEN)
@@ -22,31 +18,11 @@ def load_model():
 model = load_model()
 
 # Streamlit UI
-st.title("🤖 Open-Source Chatbot")
-st.write("A simple chatbot powered by Mistral-7B!")
-
-# Initialize chat history
-if "messages" not in st.session_state:
-    st.session_state["messages"] = []
+st.title("🤖 AI Developer Assistant")
+st.write("A chatbot powered by Mistral-7B!")
 
-# Display chat history
-for msg in st.session_state["messages"]:
-    st.chat_message(msg["role"]).write(msg["content"])
-
-# User input
-user_input = st.text_input("You:", key="user_input")
+user_input = st.text_input("You:")
 if st.button("Send"):
     if user_input:
-        st.session_state["messages"].append({"role": "user", "content": user_input})
-
-        # Display user message
-        st.chat_message("user").write(user_input)
-
-        # Get response from model
-        response_container = st.empty()
-        bot_response = model(user_input, max_length=100, do_sample=True)[0]['generated_text']
-
-        response_container.write(bot_response)
-
-        # Append bot response to chat history
-        st.session_state["messages"].append({"role": "assistant", "content": bot_response})
+        response = model(user_input, max_length=100, do_sample=True)[0]['generated_text']
+        st.write(response)