Update app.py

app.py

@@ -38,12 +38,14 @@ def save_memory(memory):
 memory = load_memory()
 
 # -----------------------
-# Chat function
+# Chat function (original behavior)
+# returns (history, history) to match previous usage
 # -----------------------
 def chat_with_model(message, history, context):
     if not isinstance(history, list):
         history = []
 
+    # prevent empty messages
     if not message:
         return history, history
 
@@ -55,29 +57,19 @@ def chat_with_model(message, history, context):
         save_memory(history)
         return history, history
 
-    # 🧠 STUDY-ONLY FILTER 🚫
-    study_keywords = [
-        "math", "science", "physics", "chemistry", "biology", "ict", "english",
-        "education", "grammar", "programming", "algorithm", "machine learning",
-        "ai", "python", "html", "css", "java", "history", "geography", "technology"
-    ]
-    if not any(word in message.lower() for word in study_keywords):
-        history.append((message, "🚫 Sorry, EduAI only answers study-related questions."))
-        return history, history
-
-    # 🧠 Conversation setup
-    conversation = [{
-        "role": "system",
-        "content": (
-            "You are EduAI — an educational AI assistant created by Wafa Fazly "
-            "You help students learn subjects such as Math, Science, English, and IT. "
-            "EduAI runs on the model 'openai/gpt-oss-safeguard-20b:groq', which was originally "
-            "trained by OpenAI. Always answer truthfully when asked about your creation. "
-            "Never chat unnecessary conversations, and don't talk anything apart from education."
-        )
-    }]
-
+    # 🧠 Build conversation
+    conversation = [{"role": "system", "content": (
+    "You are EduAI — an educational AI assistant created by Wafa Fazly "
+    "from Fathima Muslim Ladies College. "
+    "You help students learn subjects such as Math, Science, English, and IT. "
+    "EduAI runs on the model 'openai/gpt-oss-safeguard-20b:groq', which was originally "
+    "trained by OpenAI. Always answer truthfully when asked about your creation."
+    "never chat unnecessary conversations, and don't talk anything apart from education"
+    )}]
+
+    # convert tuples to messages if necessary (keeps old history format)
     for past in history[-5:]:
+        # expect (user_message, bot_reply)
         if isinstance(past, tuple) and len(past) == 2:
             conversation.append({"role": "user", "content": past[0]})
             conversation.append({"role": "assistant", "content": past[1]})
@@ -86,6 +78,7 @@ def chat_with_model(message, history, context):
 
     conversation.append({"role": "user", "content": message})
 
+    # 🚀 Send to Hugging Face model
     try:
         response = requests.post(
             "https://router.huggingface.co/v1/chat/completions",
@@ -103,16 +96,22 @@ def chat_with_model(message, history, context):
         data = response.json()
         reply = data["choices"][0]["message"]["content"]
 
+        # 🧮 Clean up math formatting (keeps your original formatting code)
         reply = reply.replace("Step", "\n\n**Step")
         reply = reply.replace(":", ":**")
         reply = reply.replace("\\[", "\n\n\\[")
         reply = reply.replace("\\]", "\\]\n\n")
 
+        if "\\" in reply or "log_" in reply or "^" in reply:
+            reply = f"{reply}"
+
         history.append((message, reply))
         save_memory(history)
+        # IMPORTANT: return a pair (history, history) because other code expects two outputs
         return history, history
 
     except Exception as e:
+        print("Error:", e)
         history.append((message, "😅 EduAI is having trouble connecting right now. Please try again later!"))
         return history, history
 
@@ -129,39 +128,49 @@ def clear_memory():
     return [], "🧹 Chat memory cleared! Start fresh."
 
 # -----------------------
-# Pause / Send wrappers (unchanged)
+# Pause / Send wrappers (FIXED)
 # -----------------------
+
+# send handler that respects paused state (ALWAYS returns (chat_history, textbox_clear))
 def send_handler(message, history, context, paused_state):
     if paused_state:
+        # Do not call the model when paused — append a friendly hint
         if not isinstance(history, list):
             history = []
         history.append((None, "⏸️ Chat is paused. Click Resume to continue."))
-        return history, ""
-    hist_pair = chat_with_model(message, history, context)
+        return history, ""  # update chatbot and clear textbox
+    # Not paused: call original chat handler and adapt its returns to (chat_history, textbox_clear)
+    hist_pair = chat_with_model(message, history, context)  # returns (history, history)
+    # unpack safely
     if isinstance(hist_pair, tuple) and len(hist_pair) == 2:
         hist = hist_pair[0]
     else:
         hist = hist_pair
     return hist, ""
 
+
+# toggle pause/resume and update UI (state + chat + button text + send button disabled)
 def toggle_pause(paused_state, history):
     new_state = not bool(paused_state)
     if not isinstance(history, list):
         history = []
 
     if new_state:
+        # now paused
         history.append((None, "⏸️ Chat paused. Send is disabled."))
         pause_btn_update = gr.Button.update(value="▶ Resume")
         send_btn_update = gr.Button.update(disabled=True)
     else:
+        # resumed
         history.append((None, "▶️ Chat resumed. You can send messages now."))
         pause_btn_update = gr.Button.update(value="⏸ Pause")
         send_btn_update = gr.Button.update(disabled=False)
 
+    # return new pause state, updated chat history, and two UI updates (pause button & send button)
     return new_state, history, pause_btn_update, send_btn_update
 
 # -----------------------
-# Build UI
+# Build UI (unchanged layout; pause added)
 # -----------------------
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
     gr.Markdown(
@@ -203,7 +212,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
                     EduAI was designed and fine-tuned by **Wafa Fazly**,  
                     a passionate Sri Lankan student 👩‍💻  
                     to help learners explore **Science, ICT, English, and more** —  
-                    in a smart and focused way! 🌟
+                    in a smart and friendly way! 🌟
                     """
                 )
 
@@ -227,14 +236,20 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
             with gr.Row():
                 send = gr.Button("✨ Send Message")
                 pause = gr.Button("⏸ Pause", variant="secondary")
+                # state to keep track of pause (False = running, True = paused)
                 pause_state = gr.State(False)
 
-    # Event handlers
+    # 🪄 Event handlers
     subj.change(update_context, inputs=subj, outputs=context_display)
     planner.change(update_context, inputs=planner, outputs=context_display)
     lang.change(update_context, inputs=lang, outputs=context_display)
+
+    # send now uses send_handler and respects pause_state; outputs: chatbot and clears textbox
     send.click(send_handler, inputs=[msg, chatbot, context_display, pause_state], outputs=[chatbot, msg])
+
     clear_btn.click(clear_memory, outputs=[chatbot, context_display])
+
+    # pause toggles pause_state, updates chatbot with a message, updates pause button label and disables/enables send
     pause.click(toggle_pause, inputs=[pause_state, chatbot], outputs=[pause_state, chatbot, pause, send])
 
-iface.launch()
+iface.launch()