Update app.py

app.py

@@ -656,12 +656,14 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                     no_key_msg = "L'Agente AI non è disponibile. Assicurati che GEMINI_API_KEY sia configurata e che l'agente sia caricato."
                     current_chat_history.append([user_message, no_key_msg])
                     # Ensure all outputs are updated
-                    return current_chat_history, current_chat_history, gr.update(value=""), gr.update(value=no_key_msg), gr.update(value="Nessuno schema disponibile.")
+                    yield current_chat_history, current_chat_history, gr.update(value=""), gr.update(value=no_key_msg), gr.update(value="Nessuno schema disponibile.")
+                    return
     
                 if not user_message.strip():
                     # Ensure all outputs are updated even for empty message
-                    return current_chat_history, current_chat_history, gr.update(value=""), gr.update(value="Stato Agente: Per favore, inserisci una domanda."), gr.update()
-    
+                    yield current_chat_history, current_chat_history, gr.update(value=""), gr.update(value="Stato Agente: Per favore, inserisci una domanda."), gr.update()
+                    return
+                
                 status_update_msg = "Stato Agente: Elaborazione della tua richiesta..."
                 pending_history = current_chat_history + [[user_message, None]]
                 # Yield intermediate state to show user message immediately
@@ -720,8 +722,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                         error_msg = "Errore: Impossibile inizializzare l'agente AI."
                         # Use pending_history which already has the user message
                         updated_history = pending_history[:-1] + [[user_message, error_msg]] # Replace None with error
-                        return updated_history, updated_history, gr.update(value=""), gr.update(value="Stato Agente: Errore di inizializzazione"), gr.update(value=schemas_text_for_display)
-                
+                        yield updated_history, updated_history, gr.update(value=""), gr.update(value="Stato Agente: Errore di inizializzazione"), gr.update(value=schemas_text_for_display)
+                        return
+                    
                     logging.info(f"Sending to EB Agent. User: '{user_message}'. DF Keys: {list(dataframes_for_agent.keys())}")
                     ai_response_dict = await current_agent.process_query(user_query=user_message)  
                     
@@ -763,16 +766,17 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                     updated_history = pending_history[:-1] + [[user_message, bot_message_for_display]]
                     
                     status_update_msg = "Stato Agente: Risposta ricevuta."
-                    return updated_history, updated_history, gr.update(value=""), gr.update(value=status_update_msg), gr.update(value=schemas_text_for_display)
-                
+                    yield updated_history, updated_history, gr.update(value=""), gr.update(value=status_update_msg), gr.update(value=schemas_text_for_display)
+                    return
+                    
                 except Exception as e:
                     logging.error(f"Error during EB Agent processing: {e}", exc_info=True)
                     error_msg_for_chat = f"# Errore dell'Agente AI:\n{type(e).__name__}: {str(e)}"
                     # Use pending_history which already has the user message
                     updated_history = pending_history[:-1] + [[user_message, error_msg_for_chat]]
                     status_update_msg = f"Stato Agente: Errore - {type(e).__name__}"
-                    return updated_history, updated_history, gr.update(value=""), gr.update(value=status_update_msg), gr.update(value=schemas_text_for_display)
-
+                    yield updated_history, updated_history, gr.update(value=""), gr.update(value=status_update_msg), gr.update(value=schemas_text_for_display)
+                    return
 
             def clear_eb_agent_chat_history():
                 initial_msg = "Ciao! Sono il tuo Agente AI per l'Employer Branding. Come posso aiutarti?" if EB_AGENT_AVAILABLE else "Agente AI non disponibile."