import streamlit as st
import os
from langchain_huggingface import HuggingFaceEndpoint

st.set_page_config(page_title="Educational Chatbot")
st.title("🎓 Educational Chatbot")

user_input = st.text_input("Ask me anything:")

if "history" not in st.session_state:
    st.session_state.history = []

if user_input:
    try:
        llm = HuggingFaceEndpoint(
            repo_id="mistralai/Mistral-7B-Instruct-v0.1",
            huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
            temperature=0.5,
            max_new_tokens=100,
        )
        response = llm.invoke(user_input)
        st.session_state.history.append(("You", user_input))
        st.session_state.history.append(("Bot", response))
    except Exception as e:
        st.error(f"⚠️ Error: {e}")

for sender, msg in reversed(st.session_state.history):
    st.markdown(f"**{sender}:** {msg}")