Auto-generate when uploaded

app_dialogue.py

@@ -921,7 +921,31 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base()) as demo:
         ],
         outputs=[textbox, generated_memes_gallery, chatbot],
     )
-    submit_btn.click(fn=lambda: "", inputs=[], outputs=[generated_memes_gallery]).then(
+    imagebox.upload(fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False).then(
+        fn=lambda: "Write a meme about this image.", inputs=[], outputs=[textbox], queue=False).then(
+        fn=model_inference,
+        inputs=[
+            model_selector,
+            system_prompt,
+            textbox,
+            chatbot,
+            imagebox,
+            decoding_strategy,
+            temperature,
+            max_new_tokens,
+            repetition_penalty,
+            top_p,
+            all_caps_meme_text,
+            text_at_the_top,
+            font_meme_text,
+        ],
+        outputs=[
+            textbox,
+            generated_memes_gallery,
+            chatbot,
+        ],
+    )
+    submit_btn.click(fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False).then(
         fn=model_inference,
         inputs=[
             model_selector,