clearing texts

app.py

@@ -26,7 +26,7 @@ def format_chat_history(chat_history):
 # Main chat handler
 def chat_with_model(model_choice, prompt, image_url, chat_history):
     if not prompt:
-        return "Please enter a text prompt.", chat_history
+        return "Please enter a text prompt.", chat_history, "", ""
 
     if chat_history is None:
         chat_history = []
@@ -67,9 +67,9 @@ def chat_with_model(model_choice, prompt, image_url, chat_history):
             chat_history.append({"role": "assistant", "content": bot_msg})
 
         else:
-            return "Unsupported model selected.", chat_history
+            return "Unsupported model selected.", chat_history, "", ""
 
-        return format_chat_history(chat_history), chat_history
+        return format_chat_history(chat_history), chat_history, "", ""
 
     except Exception as e:
         return f"Error: {e}", chat_history
@@ -100,13 +100,13 @@ with gr.Blocks() as demo:
     submit_btn.click(
         fn=chat_with_model,
         inputs=[model_dropdown, prompt_input, image_url_input, state],
-        outputs=[output_box, state]
+        outputs=[output_box, state, prompt_input, image_url_input]
     )
 
     reset_btn.click(
-        fn=lambda: ("🧹 Conversation reset. You can start a new one.", []),
+        fn=lambda: ("🧹 Conversation reset. You can start a new one.", [], "", ""),
         inputs=[],
-        outputs=[output_box, state]
+        outputs=[output_box, state, prompt_input, image_url_input]
     )
 
 demo.launch()