Update Gradio_UI.py

Changed metadata status from invalid "running" → valid "pending"/"done"

Gradio_UI.py

@@ -11,7 +11,6 @@ from smolagents.utils import _is_package_available
 def pull_messages_from_step(step_log: MemoryStep):
     """Extract ChatMessage objects from agent steps with proper nesting and error resilience"""
     import gradio as gr
-    
     if isinstance(step_log, ActionStep):
         step_number = f"Step {step_log.step_number}" if step_log.step_number is not None else "Processing"
         yield gr.ChatMessage(role="assistant", content=f"**{step_number}**")
@@ -29,32 +28,30 @@ def pull_messages_from_step(step_log: MemoryStep):
         if hasattr(step_log, "tool_calls") and step_log.tool_calls:
             tool_call = step_log.tool_calls[0]
             parent_id = f"tool_{step_log.step_number}"
-            
             args = tool_call.arguments
             if isinstance(args, dict):
                 content = "\n".join(f"• {k}: {v}" for k, v in args.items() if v and k != 'self')
             else:
                 content = str(args).strip()
-            
             metadata: Dict = {
                 "title": f"🛠️ Using: {tool_call.name}",
                 "id": parent_id,
-                "status": "running",
+                "status": "pending",  # FIXED: "running" → "pending" (Gradio validation)
             }
             yield gr.ChatMessage(role="assistant", content=content, metadata=metadata)
         
-        # Show observations/results (only include parent_id if defined)
+        # Show observations/results
         if hasattr(step_log, "observations") and step_log.observations:
             obs = step_log.observations.strip()
             if obs and not obs.startswith("Execution logs:"):
-                metadata = {"title": "✅ Result"}
+                metadata = {"title": "✅ Result", "status": "done"}  # FIXED: added status
                 if parent_id is not None:
                     metadata["parent_id"] = parent_id
                 yield gr.ChatMessage(role="assistant", content=obs, metadata=metadata)
         
-        # Show errors (only include parent_id if defined)
+        # Show errors
         if hasattr(step_log, "error") and step_log.error:
-            metadata = {"title": "⚠️ Warning"}
+            metadata = {"title": "⚠️ Warning", "status": "done"}  # FIXED: added status
             if parent_id is not None:
                 metadata["parent_id"] = parent_id
             yield gr.ChatMessage(role="assistant", content=str(step_log.error), metadata=metadata)
@@ -65,7 +62,6 @@ def pull_messages_from_step(step_log: MemoryStep):
             footer_parts.append(f"⏱️ {float(step_log.duration):.1f}s")
         if hasattr(step_log, "input_token_count") and hasattr(step_log, "output_token_count"):
             footer_parts.append(f"💬 {step_log.input_token_count + step_log.output_token_count:,} tokens")
-        
         yield gr.ChatMessage(
             role="assistant",
             content=f'<span style="color: #888; font-size: 0.85em;">{" | ".join(footer_parts)}</span>',
@@ -81,42 +77,35 @@ def stream_to_gradio(
     """Runs agent and streams messages as gradio ChatMessages with error resilience"""
     if not _is_package_available("gradio"):
         raise ModuleNotFoundError("Install gradio: `pip install 'smolagents[gradio]'`")
-    
     import gradio as gr
-    
     try:
         for step_log in agent.run(task, stream=True, reset=reset_agent_memory, additional_args=additional_args):
             if isinstance(step_log, ActionStep):
                 if hasattr(agent.model, "last_input_token_count") and hasattr(agent.model, "last_output_token_count"):
                     step_log.input_token_count = agent.model.last_input_token_count
                     step_log.output_token_count = agent.model.last_output_token_count
-                
                 for message in pull_messages_from_step(step_log):
                     yield message
-        
         final = handle_agent_output_types(step_log)
         if isinstance(final, AgentText):
             content = final.to_string()
-            yield gr.ChatMessage(role="assistant", content=content, metadata={"react": True})
+            yield gr.ChatMessage(role="assistant", content=content, metadata={"react": True, "status": "done"})
         else:
-            yield gr.ChatMessage(role="assistant", content=f"**Final Answer:** {str(final)}")
-    
+            yield gr.ChatMessage(role="assistant", content=f"**Final Answer:** {str(final)}", metadata={"status": "done"})
     except Exception as e:
         yield gr.ChatMessage(
             role="assistant",
-            content=f"⚠️ **Error during planning:** {str(e)}\n\nPlease try again with a clearer trip description (include destination, dates, origin city, and budget).",
+            content=f"⚠️ **Error during planning:** {str(e)}\nPlease try again with a clearer trip description (include destination, dates, origin city, and budget).",
+            metadata={"status": "done"}
         )
 
 class GradioUI:
     """Production-ready Gradio interface for travel agent"""
-    
     def __init__(self, agent: MultiStepAgent, file_upload_folder: Optional[str] = None):
         if not _is_package_available("gradio"):
             raise ModuleNotFoundError("Install gradio: `pip install 'smolagents[gradio]'`")
-        
         self.agent = agent
         self.file_upload_folder = file_upload_folder
-        
         if self.file_upload_folder and not os.path.exists(file_upload_folder):
             os.makedirs(file_upload_folder, exist_ok=True)
     
@@ -124,18 +113,15 @@ class GradioUI:
         import gradio as gr
         history.append(gr.ChatMessage(role="user", content=prompt))
         yield history
-        
         for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
             history.append(msg)
             yield history
     
     def launch(self, **kwargs):
         import gradio as gr
-        
         with gr.Blocks(title="✈️ Travel Catalogue Creator", fill_height=True, theme=gr.themes.Soft()) as demo:
             gr.Markdown("# ✈️ Smart Travel Catalogue Creator")
             gr.Markdown("Plan your trip with AI: weather forecasts, custom itineraries, packing lists & visual inspiration")
-            
             chatbot = gr.Chatbot(
                 label="Your Travel Assistant",
                 type="messages",
@@ -146,7 +132,6 @@ class GradioUI:
                 height=600,
                 show_copy_button=True,
             )
-            
             with gr.Row():
                 text_input = gr.Textbox(
                     label="Describe your trip",
@@ -154,13 +139,11 @@ class GradioUI:
                     lines=2,
                 )
                 submit_btn = gr.Button("Plan My Trip", variant="primary")
-            
             gr.Markdown("""
             ### 💡 Tips for best results:
-            • Include: destination, dates, origin city, budget amount + currency  
+            • Include: destination, dates, origin city, budget amount + currency
             • Example: *"5-day Barcelona trip from NYC, Oct 15-19, budget $1500 USD"*
             """)
-            
             submit_btn.click(
                 self.interact_with_agent,
                 inputs=[text_input, chatbot],
@@ -173,7 +156,6 @@ class GradioUI:
                 outputs=[chatbot],
                 show_progress="full",
             )
-        
         demo.launch(**kwargs)
 
 __all__ = ["stream_to_gradio", "GradioUI"]