Update app.py

app.py

@@ -3,7 +3,14 @@ import gradio as gr
 import requests
 import json
 
-# 🌍 Web search function
+# Test HF_TOKEN
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    print("Error: HF_TOKEN not set. Please set your Hugging Face API token.")
+else:
+    print("HF_TOKEN loaded successfully.")
+
+# 🌍 Web search function (unchanged)
 def search_web(query):
     try:
         url = "https://api.duckduckgo.com/"
@@ -20,8 +27,7 @@ def search_web(query):
     except Exception as e:
         return f"Search error: {e}"
 
-# 🧠 Memory setup
-HF_TOKEN = os.getenv("HF_TOKEN")
+# 🧠 Memory setup (unchanged)
 MEMORY_FILE = "memory.json"
 
 def load_memory():
@@ -36,12 +42,11 @@ def save_memory(memory):
 
 memory = load_memory()
 
-# 💬 Chat function
+# 💬 Chat function with better error handling
 def chat_with_model(message, history, context, file_input=None):
     if not isinstance(history, list):
         history = []
 
-    # 🌍 Web search mode
     if message.lower().startswith("search "):
         query = message[7:]
         search_result = search_web(query)
@@ -49,12 +54,10 @@ def chat_with_model(message, history, context, file_input=None):
         save_memory(history)
         return history, history
 
-    # 📂 If file is uploaded
     if file_input:
         file_name = file_input.name
         message += f"\n\n📎 (User uploaded a file named '{file_name}')"
 
-    # 🧠 Build conversation
     conversation = [
         {"role": "system", "content": (
             "You are EduAI, a multilingual educational AI assistant created by a Sri Lankan student named Wafa Fazly. "
@@ -70,8 +73,8 @@ def chat_with_model(message, history, context, file_input=None):
 
     conversation.append({"role": "user", "content": message})
 
-    # 🚀 Send to Hugging Face model
     try:
+        # Primary API call
         response = requests.post(
             "https://router.huggingface.co/v1/chat/completions",
             headers={
@@ -83,24 +86,26 @@ def chat_with_model(message, history, context, file_input=None):
                 "messages": conversation
             }
         )
+        if response.status_code != 200:
+            raise Exception(f"API Error: {response.status_code} - {response.text}")
+        
         data = response.json()
         reply = data["choices"][0]["message"]["content"]
 
-        # 🧮 Clean up math formatting
-        reply = reply.replace("Step", "\n\n**Step")
-        reply = reply.replace(":", ":**")
-        reply = reply.replace("\\[", "\n\n\\[")
-        reply = reply.replace("\\]", "\\]\n\n")
+        reply = reply.replace("Step", "\n\n**Step").replace(":", ":**").replace("\\[", "\n\n\\[").replace("\\]", "\\]\n\n")
 
         history.append((message, reply))
         save_memory(history)
         return history, history
 
     except Exception as e:
-        print("Error:", e)
-        history.append((message, "😅 EduAI is having trouble connecting right now. Please try again later!"))
+        print("Backend Error:", e)
+        error_msg = "😅 EduAI is having trouble connecting right now. Check your HF_TOKEN or try again later!"
+        history.append((message, error_msg))
         return history, history
 
+# Rest of the code (unchanged, including UI and CSS)
+
 # 📘 Sidebar context update
 def update_context(choice):
     if not choice:
@@ -113,7 +118,7 @@ def clear_memory():
         os.remove(MEMORY_FILE)
     return [], "🧹 Chat memory cleared! Start fresh."
 
-# 🖌 Custom CSS for dark theme without green, using blue accents, file uploader as symbol in text area, no send button
+# 🖌 Custom CSS (unchanged)
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;600&display=swap');
 
@@ -265,12 +270,11 @@ body {
 }
 """
 
-# 🎨 Gradio Interface with dark theme (blue accents, no green), file uploader as symbol, send on Enter (no send button)
-with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as iface:  # Using Base theme for full custom control
+# 🎨 Gradio Interface (unchanged)
+with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as iface:
     gr.Markdown("# 🎓 **EduAI — Your Professional Study Companion**", elem_classes="header-title")
 
     with gr.Row():
-        # Sidebar
         with gr.Column(scale=1, min_width=250, elem_classes="sidebar"):
             gr.Markdown("### 🧭 **Main Menu**", elem_classes="menu-title")
 
@@ -304,7 +308,6 @@ with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as iface:  # Using Base t
                 Helps learners understand **Science, ICT, English, and more** in a simple, friendly way! 💬
                 """, elem_classes="about-text")
 
-        # Main Chat Area
         with gr.Column(scale=4, elem_classes="main-chat"):
             context_display = gr.Markdown(
                 "📘 **You are in General Mode.** Ask EduAI anything about your studies!",
@@ -323,11 +326,10 @@ with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as iface:  # Using Base t
                 msg = gr.Textbox(label="Ask EduAI:", placeholder="Type your question here...", elem_classes="chat-input", show_label=False)
                 file_input = gr.File(label="📎", file_types=[".pdf", ".docx", ".png", ".jpg"], elem_classes="file-upload", show_label=False)
 
-    # 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)
-    msg.submit(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, chatbot])  # Send on Enter
+    msg.submit(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, chatbot])
     clear_btn.click(clear_memory, outputs=[chatbot, context_display])
 
 iface.launch()