Update app.py

app.py

@@ -3,43 +3,37 @@ from transformers import pipeline
 from huggingface_hub import login
 import os
 
-# 🚨 Replace this with your actual Hugging Face access token
-HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")# <-- Insert your token here securely
-
-# Login to Hugging Face Hub
+HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 login(HF_TOKEN)
 
-# App Title & Description
 st.title("🤖📟 Arduino Expert Chatbot")
-st.markdown("Get help with Arduino code, circuit diagrams, and projects.")
+st.markdown("Ask anything about Arduino: code, circuits, projects!")
 
-# Load the model securely
 @st.cache_resource
 def load_model():
     try:
-        return pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1", token=HF_TOKEN)
+        model = pipeline(
+            "text-generation",
+            model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+            token=HF_TOKEN
+        )
+        return model
     except Exception as e:
-        st.error("🚫 Failed to load the model. Using fallback model (GPT-2).")
-        return pipeline("text-generation", model="gpt2")  # fallback public model
+        st.error("❌ Failed to load Mixtral model. Falling back to GPT-2.")
+        return pipeline("text-generation", model="gpt2")
 
-# Load model once
 model = load_model()
 
-# User Input
 query = st.text_area("Ask your Arduino question here 👇", height=150)
 
-# Generate Answer
 if st.button("Get Answer"):
     if query.strip():
         with st.spinner("Thinking... 🤖"):
             try:
-                response = model(query, max_length=512, do_sample=True, temperature=0.7)
+                prompt = f"<s>[INST] {query} [/INST]"
+                response = model(prompt, max_new_tokens=512, do_sample=True, temperature=0.7)
                 st.success(response[0]['generated_text'])
             except Exception as e:
                 st.error(f"❌ Error generating response: {e}")
     else:
-        st.warning("Please enter a question about your Arduino project.")
-
-# Footer
-st.markdown("---")
-st.markdown("Made with ❤️ using Hugging Face and Streamlit")
+        st.warning("Please enter a valid question.")