import streamlit as st
from rag_utils_llamacpp import load_faiss_index, get_embedding_model, query_index, generate_answer, nettoyer_context

st.set_page_config(page_title="🎓 EduPilot", page_icon="🧠")
st.title("🎓 EduPilot (LLM local quantisé)")

# Chargement de l’index et du modèle d'embedding
index, documents = load_faiss_index()
model_embed = get_embedding_model()

# Interface utilisateur
user_input = st.text_input("Pose ta question ici :")

if user_input:
    top_docs = query_index(user_input, index, documents, model_embed)
    context = nettoyer_context("\n".join(top_docs))
    response = generate_answer(user_input, context)

    st.markdown("### ✨ Réponse d'Edu Pilot :")
    st.write(response)

st.markdown("---")
st.caption("🔹 Développé avec ❤️ par EduPilot")