Update app.py

app.py

@@ -10,108 +10,122 @@ if "xp" not in st.session_state:
     st.session_state.level = 1
 init_leaderboard()
 
-# New smarter models
-MODEL_URLS = {
-    "Mistral 7B Instruct (High quality)": "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
-    "Falcon 7B Instruct (Good Reasoning)": "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct",
-    "Falcon RW-1B (Fast Lightweight)": "https://api-inference.huggingface.co/models/tiiuae/falcon-rw-1b",
-}
-
-def query_model(api_url, prompt):
+# Model setup
+MODEL_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+
+def enhanced_query_model(topic):
     headers = {"Authorization": f"Bearer {os.environ['HF_Token']}"}
+
+    prompt = f"""
+You are a physics and mathematics AI tutor.
+
+Given the topic: "{topic}"
+
+1. Provide a simple and clear explanation suitable for high school or college students.
+2. Create 3 meaningful practice questions to test their understanding.
+3. Suggest an appropriate simulation (either a description or a real link) to help visualize the concept.
+
+Format:
+---
+Explanation:
+...
+
+Questions:
+1.
+2.
+3.
+
+Suggested Simulation:
+...
+---
+"""
+
     payload = {"inputs": prompt}
+    response = requests.post(MODEL_URL, headers=headers, json=payload)
+    result = response.json()
+
     try:
-        response = requests.post(api_url, headers=headers, json=payload)
-        result = response.json()
         if isinstance(result, list) and "generated_text" in result[0]:
             return result[0]["generated_text"]
-        return "⚠️ Unexpected response format. Try another topic."
+        else:
+            return "⚠️ Unexpected format. Try another topic."
     except Exception as e:
         return f"⚠️ API Error: {e}"
 
-def update_xp(points):
-    st.session_state.xp += points
-    if st.session_state.xp >= 100:
-        st.session_state.level += 1
-        st.session_state.xp = 0
-
-def get_simulation_embed(topic):
-    topic_map = {
+def find_simulation_link(topic):
+    sim_map = {
         "projectile motion": "https://phet.colorado.edu/sims/html/projectile-motion/latest/projectile-motion_en.html",
         "forces": "https://phet.colorado.edu/sims/html/forces-and-motion-basics/latest/forces-and-motion-basics_en.html",
-        "wave": "https://phet.colorado.edu/sims/html/waves-intro/latest/waves-intro_en.html"
+        "electric circuits": "https://phet.colorado.edu/sims/html/circuit-construction-kit-dc/latest/circuit-construction-kit-dc_en.html",
+        "waves": "https://phet.colorado.edu/sims/html/waves-intro/latest/waves-intro_en.html",
+        "optics": "https://phet.colorado.edu/sims/html/bending-light/latest/bending-light_en.html",
     }
-    for key in topic_map:
+    for key in sim_map:
         if key in topic.lower():
-            return topic_map[key]
+            return sim_map[key]
     return "https://phet.colorado.edu/en/simulations/category/physics"
 
-# --- Streamlit UI ---
+def update_xp(points):
+    st.session_state.xp += points
+    if st.session_state.xp >= 100:
+        st.session_state.level += 1
+        st.session_state.xp = 0
+
+# Streamlit UI
 st.set_page_config(page_title="CoreQuest Galaxy 🌌", page_icon="🎯")
 st.title("🎯 CoreQuest: Master Physics and Math!")
+st.caption("Learn with AI tutors, RAG-enhanced answers, XP leveling, and interactive simulations!")
 
-st.sidebar.header("🧠 Pick Your AI Model")
-selected_model_name = st.sidebar.selectbox("Choose your AI brain:", list(MODEL_URLS.keys()))
-api_url = MODEL_URLS[selected_model_name]
-
-st.sidebar.markdown("---")
+st.sidebar.header("🏆 Leaderboard and XP Tracker")
 show_leaderboard()
 
 topic = st.text_input("📘 Enter a Physics or Math Topic:")
 
 if topic:
-    with st.spinner("Asking the galaxy..."):
-        prompt = f"Explain '{topic}' for a high school student in a simple way and generate 3 practice questions."
-        output = query_model(api_url, prompt)
+    with st.spinner("Generating deep explanations and simulations..."):
+        output = enhanced_query_model(topic)
 
     st.markdown("### 🤖 AI Response")
     st.write(output)
 
     update_xp(10)
     st.progress(st.session_state.xp)
-    st.write(f"🏆 Level {st.session_state.level} | XP: {st.session_state.xp} points")
+    st.write(f"🏆 Level {st.session_state.level} | XP: {st.session_state.xp} Points")
 
     st.markdown("---")
-    st.subheader("🔧 Actions")
+    st.subheader("🔭 Simulation Area")
+
+    sim_url = find_simulation_link(topic)
+    st.components.v1.iframe(sim_url, height=500)
 
-    if st.button("💬 Ask Standard AI"):
-        user_question = st.text_input("Your follow-up question:", key="chat_input")
-        if user_question:
-            with st.spinner("Thinking..."):
-                answer = query_model(api_url, f"Answer this: {user_question}")
-            st.success(answer)
-            update_xp(5)
+    st.markdown("---")
+    st.subheader("🔧 Actions")
 
-    if st.button("🌌 Ask Learning Galaxy GPT (with Notes)"):
-        galaxy_question = st.text_input("Ask Learning Galaxy:", key="galaxy_input")
+    if st.button("🌌 Ask Learning Galaxy GPT"):
+        galaxy_question = st.text_input("Ask a more advanced question:", key="galaxy_input")
         if galaxy_question:
             with st.spinner("Learning Galaxy thinking..."):
                 galaxy_answer = learning_galaxy_answer(galaxy_question)
             st.success(galaxy_answer)
             update_xp(10)
 
-    if st.button("🔭 View Simulation"):
-        sim_url = get_simulation_embed(topic)
-        st.components.v1.iframe(sim_url, height=500)
-        update_xp(5)
-
-    if st.button("🎁 See Hint"):
-        st.info("Hint: Sketch diagrams or list basic principles related to the topic!")
+    if st.button("🎁 See a Hint"):
+        st.info("Hint: Sketch diagrams, list formulas, or visualize forces/fields.")
         update_xp(3)
 
-    if st.button("🔄 Get More Practice"):
-        with st.spinner("Getting more practice..."):
-            new_output = query_model(api_url, prompt)
+    if st.button("🔄 Get Another Set of Practice"):
+        with st.spinner("Loading more questions..."):
+            new_output = enhanced_query_model(topic)
         st.write(new_output)
         update_xp(7)
 
     st.markdown("---")
-    st.subheader("🏅 Submit Your Score")
+    st.subheader("🏅 Submit Your XP Score")
     player_name = st.text_input("👤 Enter your name to save XP:")
     if st.button("🏆 Save to Leaderboard"):
         update_leaderboard(player_name, st.session_state.xp)
-        st.success("Score submitted successfully!")
+        st.success("XP submitted!")
         show_leaderboard()
 
 st.markdown("---")
-st.caption("Created with ❤️ | CoreQuest Galaxy 🚀 Powered by Hugging Face + Learning Galaxy GPT ✨")
+st.caption("Created with ❤️ by CoreQuest Galaxy 🚀 Powered by Hugging Face + Learning Galaxy ✨")