Update app.py

app.py

@@ -11,6 +11,12 @@ import os
 api_key = os.getenv("YOUR_OPENAI_KEY")
 openai.api_key = api_key
 
+examples = [
+    ["Explain something dark but unknown about space"],
+    ["Generate me a funny meme about friends"],
+    ["Create a caption for a sad meme about exams"]
+]
+
 # Function to generate meme caption using GPT model
 def generate_meme_caption(topic):
     response = openai.ChatCompletion.create(
@@ -81,14 +87,20 @@ with gr.Blocks() as demo:
         prompt = gr.Textbox(label="Enter your meme prompt:")
         run_button = gr.Button("Generate Meme")
         
-        result = gr.Plot(label="Generated Meme")
+        result = gr.Image(label="Generated Meme", show_label=False)
+        download_button = gr.File(label="Download Meme")  # Thêm nút download
         
+        # Function to call Gradio infer
         def gradio_infer(prompt):
             plt.figure(figsize=(10, 8))
-            create_meme(prompt)
-            plt.savefig('/tmp/meme.png')  # Lưu hình tạm thời
-            return plt
+            combined_img, output_path = create_meme(prompt)
+            return combined_img, output_path  # Trả về ảnh và đường dẫn file tải xuống
+            
+        gr.Examples(
+            examples=examples,
+            inputs=[prompt]
+        )
         
-        run_button.click(gradio_infer, inputs=[prompt], outputs=[result])
+        run_button.click(gradio_infer, inputs=[prompt], outputs=[result, download_button])  # Thêm download_button vào outputs
 
 demo.launch()