Update app.py

app.py

@@ -494,7 +494,7 @@ database = {k.lower(): v for k, v in database.items()}
 
 
 # Mock data function for chatbot
-def send_message(message, history):
+"""def send_message(message, history):
     #if not message:
     #    return history
     #history.append({"role": "assistant", "content": f"This is a response about: {message}"})
@@ -514,27 +514,33 @@ def send_message(message, history):
             continue
         else:
             partial += t
-            yield history + [{"role": "assistant", "content": partial}]    
+            yield history + [{"role": "assistant", "content": partial}]"""    
     
-
-def start_generation(message, history):
+def send_message(message, history):
     history.append({"role": "user", "content": message})
-    return message, history  # Pass message along with updated history
-
-
-def continue_generation(message, history):
+    
+    # Recupera contesto extra se esiste
     if message.lower() in database:
         context = random.choice(database[message.lower()])
         message += " \n" + context
+
     response_generator = wrapper.generate(message, history)
     partial = ""
     idx = 0
+    cleared = False
+
     for t in response_generator:
         if idx <= 3:
             idx += 1
             continue
-        partial += t
-        yield history + [{"role": "assistant", "content": partial}]
+        else:
+            partial += t
+            # Quando arriva il primo token utile, svuota la textbox
+            if not cleared:
+                cleared = True
+                yield gr.update(value=""), history + [{"role": "assistant", "content": partial}]
+            else:
+                yield history + [{"role": "assistant", "content": partial}]
 
 
 # Create the dashboard
@@ -610,31 +616,15 @@ with gr.Blocks(theme=ArtemisiaTheme(), css=custom_css) as demo:
     )
     """
     send_button.click(
-        fn=lambda msg: (msg, gr.update(value="")),
-        inputs=[chat_input],
-        outputs=[gr.State(), chat_input]  # Store original msg in State
-    ).then(
-        fn=start_generation,
-        inputs=[gr.State(), chatbot],     # Use the stored message
-        outputs=[gr.State(), chatbot]     # Store again for streaming
-    ).then(
-        fn=continue_generation,
-        inputs=[gr.State(), chatbot],     # Stream using same message
-        outputs=[chatbot]
+        fn=send_message,
+        inputs=[chat_input, chatbot],
+        outputs=[chat_input, chatbot]  # la textbox viene aggiornata qui dentro
     )
     
     chat_input.submit(
-        fn=lambda msg: (msg, gr.update(value="")),
-        inputs=[chat_input],
-        outputs=[gr.State(), chat_input]
-    ).then(
-        fn=start_generation,
-        inputs=[gr.State(), chatbot],
-        outputs=[gr.State(), chatbot]
-    ).then(
-        fn=continue_generation,
-        inputs=[gr.State(), chatbot],
-        outputs=[chatbot]
+        fn=send_message,
+        inputs=[chat_input, chatbot],
+        outputs=[chat_input, chatbot]
     )
     # JavaScript for UI enhancements
     gr.HTML("""