Update space

app.py

@@ -12,8 +12,7 @@ data = []
 def respond(message, history, system_message):
     messages = [{"role": "system", "content": system_message}]
     for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
+        messages.append({"role": "user", "content": val[0]})
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
             
@@ -77,28 +76,26 @@ def extract_table(url):
     except Exception as e:
         return f"<p>Error: {str(e)}</p>"
 
-def add_text(history, text):
-    history = history + [(text, None)]
+def user_message(message, history):
+    history = history + [(message, None)]
     return history
 
-def generate_response(history, system_message):
-    if not history:
-        return history
-    
+def bot(history, system_message):
+    history_with_response = history.copy()
     response = ""
     for chunk in respond(history[-1][0], history[:-1], system_message):
         response = chunk
-        history[-1] = (history[-1][0], response)
-        yield history
+        history_with_response[-1] = (history[-1][0], response)
+        yield history_with_response
 
 def handle_prepare(index, history):
     try:
         index = int(index)
         if 0 <= index < len(data):
             topic = data[index]["Topic"]
-            date = data[index]["Date"]
-            message = f"Please prepare a 10-minute reading guide for the topic '{topic}' scheduled for {date}"
-            return history + [(message, None)]
+            message = f"Prepare a 10-minute reading for the topic: {topic}"
+            history = history + [(message, None)]
+            return history
     except:
         return history
 
@@ -122,7 +119,7 @@ with gr.Blocks() as demo:
             
         with gr.Column(scale=2):
             chatbot = gr.Chatbot()
-            msg = gr.Textbox(label="Message")
+            msg = gr.Textbox(label="Message", interactive=True)
             system_message = gr.Textbox(
                 value="Student class preparation companion.",
                 label="System message"
@@ -140,41 +137,28 @@ with gr.Blocks() as demo:
     )
     
     # Submit button handler
-    msg.submit(
-        fn=add_text,
-        inputs=[chatbot, msg],
-        outputs=chatbot
-    ).success(
-        fn=lambda: "",
-        outputs=msg
-    ).then(
-        fn=generate_response,
-        inputs=[chatbot, system_message],
-        outputs=chatbot
-    )
-    
     submit.click(
-        fn=add_text,
-        inputs=[chatbot, msg],
-        outputs=chatbot
-    ).success(
-        fn=lambda: "",
-        outputs=msg
+        fn=user_message,
+        inputs=[msg, chatbot],
+        outputs=chatbot,
+    ).then(
+        fn=lambda: "",  # Clear input textbox
+        outputs=msg,
     ).then(
-        fn=generate_response,
+        fn=bot,
         inputs=[chatbot, system_message],
-        outputs=chatbot
+        outputs=chatbot,
     )
     
     # Prepare button handler
     prepare_button.click(
         fn=handle_prepare,
         inputs=[prepare_topic, chatbot],
-        outputs=chatbot
+        outputs=chatbot,
     ).then(
-        fn=generate_response,
+        fn=bot,
         inputs=[chatbot, system_message],
-        outputs=chatbot
+        outputs=chatbot,
     )
     
     # Clear button handler