Update app.py

app.py

@@ -11,13 +11,8 @@ 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
+# Function to generate meme caption using GPT4o mini model
+#GPT 4o-mini is cheap
 def generate_meme_caption(topic):
     response = openai.ChatCompletion.create(
         model="gpt-4o-mini",
@@ -87,20 +82,14 @@ with gr.Blocks() as demo:
         prompt = gr.Textbox(label="Enter your meme prompt:")
         run_button = gr.Button("Generate Meme")
         
-        result = gr.Image(label="Generated Meme", show_label=False)
-        download_button = gr.File(label="Download Meme")  # Thêm nút download
+        result = gr.Plot(label="Generated Meme")
         
-        # Function to call Gradio infer
         def gradio_infer(prompt):
             plt.figure(figsize=(10, 8))
-            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]
-        )
+            create_meme(prompt)
+            plt.savefig('/tmp/meme.png')  # Lưu hình tạm thời
+            return plt
         
-        run_button.click(gradio_infer, inputs=[prompt], outputs=[result, download_button])  # Thêm download_button vào outputs
+        run_button.click(gradio_infer, inputs=[prompt], outputs=[result])
 
-demo.launch()
+demo.launch()