Update app.py

app.py

@@ -152,24 +152,22 @@ agent.prompt_templates["system_prompt"] += """
     - You must use this answer in final_answer
 """
 
-def set_loading():
-    return "⏳ Jerry is working on it..."
-
-def clear_loading():
-    return ""
-
 def run_agent(query, nsfw_detection_input):
     global image_output
     image_output = None
-    
+
+    yield None, "⏳ Jerry is thinking... please wait"
+
     try:
         response = agent.run(
-            query if query else "", 
+            query if query else "",
             additional_args={"nsfw_detection_input": nsfw_detection_input}
         )
-        return image_output, str(response)
+
+        yield image_output, str(response)
+
     except Exception as e:
-        return None, f"Agent Error: {str(e)}"
+        yield None, f"❌ Agent Error: {str(e)}"
 
 with gr.Blocks(title="Jerry AI Assistant") as demo:
     gr.Markdown("# 🤖 Jerry - Your AI Assistant")