Create app.py

app.py

@@ -0,0 +1,204 @@
+import os
+import base64
+import zipfile
+import tempfile
+import uuid
+import json
+import requests
+from datetime import datetime
+from PIL import Image
+import gradio as gr
+from io import BytesIO
+
+# Persistent storage setup
+BASE_DIR = os.path.abspath(os.getcwd())
+HISTORY_DIR = os.path.join(BASE_DIR, "image_history")
+os.makedirs(HISTORY_DIR, exist_ok=True)
+ZIP_PATH = os.path.join(BASE_DIR, "image_history.zip")
+
+# Store metadata
+HISTORY_METADATA = os.path.join(HISTORY_DIR, "metadata.json")
+if not os.path.exists(HISTORY_METADATA):
+    with open(HISTORY_METADATA, "w") as f:
+        json.dump([], f)
+
+def load_metadata():
+    with open(HISTORY_METADATA, "r") as f:
+        return json.load(f)
+
+def save_metadata(metadata):
+    with open(HISTORY_METADATA, "w") as f:
+        json.dump(metadata, f, indent=2)
+
+def save_image(image: Image.Image, username, prompt):
+    filename = os.path.join(HISTORY_DIR, f"{uuid.uuid4()}.png")
+    image.save(filename)
+    metadata = load_metadata()
+    metadata.append({
+        "path": filename,
+        "username": username,
+        "prompt": prompt,
+        "timestamp": datetime.now().isoformat()
+    })
+    save_metadata(metadata)
+    return filename
+
+def get_history(username=None):
+    metadata = load_metadata()
+    if username:
+        metadata = [m for m in metadata if m["username"] == username]
+    else:
+        return [], []
+    return [(m["path"], f"{m['username']} - {m['timestamp']}") for m in metadata], [(m["path"], f"{m['username']} - {m['timestamp']}") for m in metadata]
+
+def download_history():
+    with zipfile.ZipFile(ZIP_PATH, "w") as zipf:
+        for root, dirs, files in os.walk(HISTORY_DIR):
+            for file in files:
+                if file != "metadata.json":
+                    zipf.write(os.path.join(root, file), arcname=file)
+        zipf.write(HISTORY_METADATA, arcname="metadata.json")
+    return ZIP_PATH
+
+def generate_image(prompt, bytez_api_key, aspect_ratio="1:1"):
+    api_key = bytez_api_key.strip() if bytez_api_key and bytez_api_key.strip() != "" else os.environ.get("BYTEZ_API_KEY")
+    
+    if not api_key:
+        raise ValueError("No API key provided. Set BYTEZ_API_KEY environment variable or enter in UI.")
+    
+    headers = {
+        "Authorization": api_key,
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "text": prompt,
+        "aspect_ratio": aspect_ratio
+    }
+
+    try:
+        response = requests.post(
+            "https://api.bytez.com/models/v2/google/imagen-4.0-fast-generate-001",
+            headers=headers,
+            json=payload,
+            timeout=45
+        )
+        response.raise_for_status()
+        
+        result = response.json()
+        if not result.get("results") or not isinstance(result["results"], list) or len(result["results"]) == 0:
+            raise ValueError("No results returned from API")
+        
+        # Decode base64 image
+        img_data = base64.b64decode(result["results"][0]["base64"])
+        img = Image.open(BytesIO(img_data))
+        return img
+    
+    except requests.exceptions.RequestException as e:
+        error_msg = f"API request failed: {str(e)}"
+        if hasattr(e, 'response') and e.response is not None:
+            error_msg += f"\nStatus: {e.response.status_code}\nResponse: {e.response.text}"
+        raise Exception(error_msg)
+    except Exception as e:
+        raise Exception(f"Image generation failed: {str(e)}")
+
+def process_prompt(prompt, bytez_api_key, username, aspect_ratio):
+    if not username.strip():
+        raise ValueError("Username cannot be empty")
+    
+    # Generate image using Imagen API
+    result_img = generate_image(prompt, bytez_api_key, aspect_ratio)
+    
+    # Save to history
+    saved_path = save_image(result_img, username, prompt)
+    
+    # Return for gallery display
+    return Image.open(saved_path), [(saved_path, f"{username}: {prompt}")]
+
+with gr.Blocks() as demo:
+    gr.HTML("""
+    <div style='display: flex; align-items: center; justify-content: center; gap: 20px'>
+        <div style="background-color: var(--block-background-fill); border-radius: 8px">
+            <img src="https://api.bytez.com/favicon.ico" style="width: 100px; height: 100px;">
+        </div>
+        <div>
+            <h1>Imagen 4.0 Fast Generator</h1>
+            <p>Text-to-image generation using Google's Imagen 4.0 model</p>
+        </div>
+    </div>
+    """)
+
+    with gr.Row():
+        with gr.Column():
+            bytez_api_key = gr.Textbox(
+                lines=1, 
+                placeholder="Enter Bytez API Key (optional)", 
+                label="Bytez API Key",
+                type="password"
+            )
+            username_input = gr.Textbox(
+                lines=1, 
+                placeholder="Enter your username", 
+                label="Username",
+                info="Required for saving history"
+            )
+            prompt_input = gr.Textbox(
+                lines=3, 
+                placeholder="A Raspberry Pi SBC on a wooden desk with electronic components", 
+                label="Prompt",
+                info="Be descriptive for best results"
+            )
+            aspect_ratio = gr.Dropdown(
+                choices=["1:1", "16:9", "9:16"],
+                value="1:1",
+                label="Aspect Ratio"
+            )
+            submit_btn = gr.Button("✨ Generate Image", variant="primary")
+
+        with gr.Column():
+            output_image = gr.Image(
+                label="Generated Image",
+                interactive=False
+            )
+            history_gallery = gr.Gallery(
+                label="Your History",
+                columns=2,
+                preview=True
+            )
+            with gr.Row():
+                download_zip_btn = gr.Button("📥 Download History ZIP")
+                download_zip_file = gr.File(label="Download ZIP")
+
+    # Event handlers
+    submit_btn.click(
+        fn=process_prompt,
+        inputs=[prompt_input, bytez_api_key, username_input, aspect_ratio],
+        outputs=[output_image, history_gallery],
+        api_name="generate"
+    )
+    
+    username_input.change(
+        fn=get_history,
+        inputs=username_input,
+        outputs=[history_gallery, history_gallery]
+    )
+    
+    download_zip_btn.click(
+        fn=download_history,
+        outputs=download_zip_file
+    )
+
+# Add examples section
+gr.Examples(
+    examples=[
+        ["A cyberpunk cat wearing neon goggles, digital art style", "1:1"],
+        ["Mountain landscape at sunset with northern lights, photorealistic", "16:9"],
+        ["Steampunk airship floating above Victorian city, detailed illustration", "9:16"],
+        ["Minimalist logo of a bird made from geometric shapes, blue and white", "1:1"]
+    ],
+    inputs=[prompt_input, aspect_ratio],
+    label="💡 Example Prompts"
+)
+
+if __name__ == "__main__":
+    demo.launch()