Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import base64
+import time
 from PIL import Image
 import io
 import os
@@ -9,6 +10,7 @@ API_KEY = os.environ.get("API_KEY")
 API_URL = "https://api.4llm.com/v1/images/generations"
 
 def generate_image(prompt, size="1024x1024"):
+    start_time = time.time()  # Start timing
     headers = {
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
@@ -32,12 +34,20 @@ def generate_image(prompt, size="1024x1024"):
         
         # Convert to PIL Image
         image = Image.open(io.BytesIO(image_bytes))
-        return image, result["data"][0]["revised_prompt"]
+        
+        generation_time = time.time() - start_time  # Calculate generation time
+        return image, result["data"][0]["revised_prompt"], f"{generation_time:.2f} seconds"
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        return None, f"Error: {str(e)}", ""
 
 # Create the Gradio interface
-with gr.Blocks(title="4LLM Image Generation", css="footer { display: none !important; }") as demo:
+with gr.Blocks(title="4LLM Image Generation", css="""
+    footer { display: none !important; }
+    .gr-button { background-color: #4CAF50; color: white; border: none; padding: 10px 20px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 5px; }
+    .gr-button:hover { background-color: #45a049; }
+    .gr-textbox { margin-bottom: 10px; }
+    .gr-markdown { margin-bottom: 20px; }
+""") as demo:
     gr.Markdown("# 4LLM Image Generation Demo")
     gr.Markdown("Generate images using the 4LLM API with no rate limits!")
     
@@ -53,7 +63,7 @@ with gr.Blocks(title="4LLM Image Generation", css="footer { display: none !impor
                 value="1024x1024",
                 label="Image Size"
             )
-            generate_btn = gr.Button("Generate Image")
+            generate_btn = gr.Button("🔄 Generate Image")
         
         with gr.Column():
             output_image = gr.Image(label="Generated Image")
@@ -62,12 +72,29 @@ with gr.Blocks(title="4LLM Image Generation", css="footer { display: none !impor
                 lines=3,
                 interactive=False
             )
+            generation_time_display = gr.Textbox(
+                label="Generation Time",
+                lines=1,
+                interactive=False
+            )
+            copy_btn = gr.Button("📋 Copy Image")
+            download_btn = gr.Button("⬇️ Download Image")
     
     generate_btn.click(
         fn=generate_image,
         inputs=[prompt, size],
-        outputs=[output_image, revised_prompt]
+        outputs=[output_image, revised_prompt, generation_time_display]
     )
 
+    # Add functionality for the Copy and Download buttons
+    def copy_image(image):
+        return image
+
+    def download_image(image):
+        return image
+
+    copy_btn.click(fn=copy_image, inputs=output_image, outputs=None)
+    download_btn.click(fn=download_image, inputs=output_image, outputs=None)
+
 if __name__ == "__main__":
     demo.launch(share=False)