init frontend

phase/Student_view/chatbot.py

@@ -1,94 +1,97 @@
+# phase/Student_view/chatbot.py
 import streamlit as st
 import datetime
 import os
-from openai import OpenAI
+from huggingface_hub import InferenceClient
 
 # Read from Space Secrets (Settings → Variables & secrets)
-api_key = os.getenv("OPENAI_API_KEY")
-if not api_key:
-    st.error("⚠️ OPENAI_API_KEY is not set. In your Space, add a Secret named OPENAI_API_KEY ")
-client = OpenAI(api_key=api_key)
+HF_TOKEN = os.getenv("HF_TOKEN")
+GEN_MODEL = os.getenv("GEN_MODEL", "mistralai/Mistral-7B-Instruct-v0.2")
+
+if not HF_TOKEN:
+    st.error("⚠️ HF_TOKEN is not set. In your Space, add a Secret named HF_TOKEN.")
+client = InferenceClient(model=GEN_MODEL, token=HF_TOKEN)
 
-# --- Helper to add message ---
 def add_message(text: str, sender: str):
     st.session_state.messages.append(
         {
             "id": str(datetime.datetime.now().timestamp()),
             "text": text,
-            "sender": sender,
+            "sender": sender,  # "user" or "assistant"
             "timestamp": datetime.datetime.now()
         }
     )
 
+def _reply_with_hf():
+    # Convert chat history into a simple prompt
+    system = "You are a kind Jamaican primary-school finance tutor. Keep answers short, friendly, and age-appropriate."
+    convo = []
+    for m in st.session_state.messages:
+        role = "USER" if m["sender"] != "assistant" else "ASSISTANT"
+        convo.append(f"{role}: {m['text']}")
+    prompt = (system + "\n\n" + "\n".join(convo) + "\nASSISTANT:").strip()
+
+    return client.text_generation(
+        prompt,
+        max_new_tokens=220,
+        temperature=0.2,
+        stream=False
+    )
+
 def show_page():
     st.title("🤖 AI Financial Tutor")
     st.caption("Get personalized help with your financial questions")
 
-    # --- Initialize state ---
     if "messages" not in st.session_state:
-        st.session_state.messages = [
-            {
-                "id": "1",
-                "text": "Hi! I'm your AI Financial Tutor. I'm here to help you learn about personal finance, investing, budgeting, and more. What would you like to know?",
-                "sender": "assistant",
-                "timestamp": datetime.datetime.now()
-            }
-        ]
+        st.session_state.messages = [{
+            "id": "1",
+            "text": "Hi! I'm your AI Financial Tutor. What would you like to learn today?",
+            "sender": "assistant",
+            "timestamp": datetime.datetime.now()
+        }]
     if "is_typing" not in st.session_state:
         st.session_state.is_typing = False
 
-    # --- Chat Container ---
     chat_container = st.container()
     with chat_container:
         for msg in st.session_state.messages:
             if msg["sender"] == "assistant":
-                st.markdown(f"<div style='background-color:#e0e0e0; color:black; padding:10px; border-radius:12px; max-width:70%; margin-bottom:5px;'>{msg['text']}<br><sub>{msg['timestamp'].strftime('%H:%M')}</sub></div>", unsafe_allow_html=True)
+                st.markdown(
+                    f"<div style='background-color:#e0e0e0; color:black; padding:10px; border-radius:12px; max-width:70%; margin-bottom:5px;'>{msg['text']}<br><sub>{msg['timestamp'].strftime('%H:%M')}</sub></div>",
+                    unsafe_allow_html=True
+                )
             else:
-                st.markdown(f"<div style='background-color:#4CAF50; color:white; padding:10px; border-radius:12px; max-width:70%; margin-left:auto; margin-bottom:5px;'>{msg['text']}<br><sub>{msg['timestamp'].strftime('%H:%M')}</sub></div>", unsafe_allow_html=True)
+                st.markdown(
+                    f"<div style='background-color:#4CAF50; color:white; padding:10px; border-radius:12px; max-width:70%; margin-left:auto; margin-bottom:5px;'>{msg['text']}<br><sub>{msg['timestamp'].strftime('%H:%M')}</sub></div>",
+                    unsafe_allow_html=True
+                )
 
         if st.session_state.is_typing:
             st.markdown("🤖 _FinanceBot is typing..._")
 
-    # --- Quick Questions ---
     if len(st.session_state.messages) == 1:
         st.markdown("Try asking about:")
         cols = st.columns(2)
-        quick_questions = [
+        quick = [
             "How does compound interest work?",
             "How much should I save for emergencies?",
             "What's a good budgeting strategy?",
             "How do I start investing?"
         ]
-        for i, q in enumerate(quick_questions):
+        for i, q in enumerate(quick):
             if cols[i % 2].button(q):
-                add_message(q, "user")
-                st.session_state.is_typing = True
-                st.rerun()
+                add_message(q, "user"); st.session_state.is_typing = True; st.rerun()
 
-    # --- Input Box ---
     user_input = st.chat_input("Ask me anything about personal finance...")
     if user_input:
         add_message(user_input, "user")
         st.session_state.is_typing = True
         st.rerun()
 
-    # --- Call OpenAI API directly ---
     if st.session_state.is_typing:
         try:
             with st.spinner("FinanceBot is thinking..."):
-                messages_payload = [
-                    {"role": m['sender'], "content": m['text']} for m in st.session_state.messages
-                ]
-                # Ensure role is either 'user' or 'assistant'
-                for m in messages_payload:
-                    if m['role'] not in ['user', 'assistant']:
-                        m['role'] = 'user'
-
-                response = client.chat.completions.create(
-                    model="gpt-4o-mini",
-                    messages=messages_payload
-                )
-                bot_reply = response.choices[0].message.content
+                bot_reply = _reply_with_hf()
                 add_message(bot_reply, "assistant")
         except Exception as e:
             add_message(f"⚠️ Error: {e}", "assistant")
@@ -96,7 +99,6 @@ def show_page():
             st.session_state.is_typing = False
             st.rerun()
 
-    # --- Navigation Button ---
     if st.button("Back to Dashboard", key="ai_tutor_back_btn"):
         st.session_state.current_page = "Student Dashboard"
         st.rerun()

requirements.txt

@@ -31,3 +31,8 @@ markdownify>=0.12.1
 duckduckgo-search==6.3.5
 bcrypt
 plotly
+
+
+python-dotenv==1.0.1
+requests>=2.32
+huggingface_hub==0.24.5

utils/api.py

@@ -1,6 +1,10 @@
+# utils/api.py
 import os, requests
 
-BACKEND = os.getenv("BACKEND_URL").rstrip("/")
+BACKEND = (os.getenv("BACKEND_URL") or "").strip()
+if not BACKEND:
+    raise RuntimeError("BACKEND_URL is not set in Space secrets.")
+BACKEND = BACKEND.rstrip("/")
 
 def start_agent(student_id:int, lesson_id:int, level_slug:str):
     r = requests.post(f"{BACKEND}/agent/start", json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug})