Update app.py

app.py

@@ -316,12 +316,7 @@ def answer_question(message, history, mode, language_mode, quiz_count_mode, show
         ensure_loaded()
         user_text = message.strip()
 
-        # Add user message first
-        history.append({"role": "user", "content": user_text})
-
-        # -----------------------------------------
-        # QUIZ EVALUATION STEP
-        # -----------------------------------------
+        # Quiz evaluation step
         if quiz_state.get("active", False):
             evaluation_prompt = build_quiz_evaluation_prompt(
                 quiz_state["language_mode"],
@@ -345,7 +340,7 @@ def answer_question(message, history, mode, language_mode, quiz_count_mode, show
                 lines.append(f"**Feedback:** {item['feedback']}")
 
             final_answer = "\n".join(lines)
-            history.append({"role": "assistant", "content": final_answer})
+            history.append((user_text, final_answer))
 
             quiz_state = {
                 "active": False,
@@ -356,15 +351,11 @@ def answer_question(message, history, mode, language_mode, quiz_count_mode, show
 
             return history, quiz_state, ""
 
-        # -----------------------------------------
-        # NORMAL RETRIEVAL
-        # -----------------------------------------
+        # Normal retrieval
         records = search_hybrid(user_text, shortlist_k=20, final_k=3)
         context = build_context(records)
 
-        # -----------------------------------------
-        # QUIZ GENERATION
-        # -----------------------------------------
+        # Quiz generation
         if mode == "Quiz Me":
             n_questions = choose_quiz_count(user_text, quiz_count_mode)
             prompt = build_quiz_generation_prompt(language_mode, user_text, context, n_questions)
@@ -389,7 +380,7 @@ def answer_question(message, history, mode, language_mode, quiz_count_mode, show
                 lines.append(make_sources(records))
 
             assistant_text = "\n".join(lines)
-            history.append({"role": "assistant", "content": assistant_text})
+            history.append((user_text, assistant_text))
 
             quiz_state = {
                 "active": True,
@@ -400,20 +391,18 @@ def answer_question(message, history, mode, language_mode, quiz_count_mode, show
 
             return history, quiz_state, ""
 
-        # -----------------------------------------
-        # OTHER MODES
-        # -----------------------------------------
+        # Other modes
         prompt = build_tutor_prompt(mode, language_mode, user_text, context)
         answer = chat_text(prompt)
 
         if show_sources:
             answer += "\n\n---\n**Sources used:**\n" + make_sources(records)
 
-        history.append({"role": "assistant", "content": answer})
+        history.append((user_text, answer))
         return history, quiz_state, ""
 
     except Exception as e:
-        history.append({"role": "assistant", "content": f"Error: {str(e)}"})
+        history.append((message, f"Error: {str(e)}"))
         quiz_state["active"] = False
         return history, quiz_state, ""
 
@@ -457,7 +446,7 @@ footer { display: none !important; }
 """
 
 
-with gr.Blocks(css=CSS) as demo:
+with gr.Blocks() as demo:
     quiz_state = gr.State({
         "active": False,
         "topic": None,
@@ -495,7 +484,7 @@ with gr.Blocks(css=CSS) as demo:
 - The system will ask questions, and your **next message will be evaluated automatically**
 """)
 
-    chatbot = gr.Chatbot(height=520, type="messages")
+    chatbot = gr.Chatbot(height=520)
     msg = gr.Textbox(
         placeholder="Ask a question or type a topic...",
         lines=1,
@@ -525,4 +514,4 @@ with gr.Blocks(css=CSS) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css=CSS)