vineet124jig's picture
Update app.py
a471a75 verified
raw
history blame
11.1 kB
import gradio as gr
import requests
import json
import os
import time
from collections import defaultdict
from PIL import Image
import io
BASE_URL = "https://api.jigsawstack.com/v1"
headers = {
"x-api-key": os.getenv("JIGSAWSTACK_API_KEY")
}
# Rate limiting configuration
request_times = defaultdict(list)
MAX_REQUESTS = 1 # Maximum requests per time window
TIME_WINDOW = 3600 # Time window in seconds (1 hour)
def get_real_ip(request: gr.Request):
"""Extract real IP address using x-forwarded-for header or fallback"""
if not request:
return "unknown"
forwarded = request.headers.get("x-forwarded-for")
if forwarded:
ip = forwarded.split(",")[0].strip() # First IP in the list is the client's
else:
ip = request.client.host # fallback
return ip
def check_rate_limit(request: gr.Request):
"""Check if the current request exceeds rate limits"""
if not request:
return True, "Rate limit check failed - no request info"
ip = get_real_ip(request)
now = time.time()
# Clean up old timestamps outside the time window
request_times[ip] = [t for t in request_times[ip] if now - t < TIME_WINDOW]
# Check if rate limit exceeded
if len(request_times[ip]) >= MAX_REQUESTS:
time_remaining = int(TIME_WINDOW - (now - request_times[ip][0]))
time_remaining_minutes = round(time_remaining / 60, 1)
time_window_minutes = round(TIME_WINDOW / 60, 1)
return False, f"Rate limit exceeded. You can make {MAX_REQUESTS} requests per {time_window_minutes} minutes. Try again in {time_remaining_minutes} minutes."
# Add current request timestamp
request_times[ip].append(now)
return True, ""
def generate_image(prompt, aspect_ratio, width, height, steps, negative_prompt, guidance, seed, request: gr.Request):
rate_limit_ok, rate_limit_msg = check_rate_limit(request)
if not rate_limit_ok:
return None, {"error": "Rate limit exceeded", "message": rate_limit_msg}
# Validate required inputs
if not prompt or not prompt.strip():
return None, {"error": "Prompt is required"}
if len(prompt.strip()) > 5000:
return None, {"error": "Prompt must be between 1-5000 characters"}
# Build payload with required and optional parameters
payload = {
"prompt": prompt.strip()
}
# Add optional parameters if provided
if aspect_ratio and aspect_ratio.strip():
payload["aspect_ratio"] = aspect_ratio.strip()
if width:
try:
width_int = int(width)
if 256 <= width_int <= 1920:
payload["width"] = width_int
else:
return None, {"error": "Width must be between 256-1920 pixels"}
except ValueError:
return None, {"error": "Width must be a valid number"}
if height:
try:
height_int = int(height)
if 256 <= height_int <= 1920:
payload["height"] = height_int
else:
return None, {"error": "Height must be between 256-1920 pixels"}
except ValueError:
return None, {"error": "Height must be a valid number"}
if steps:
try:
steps_int = int(steps)
if 1 <= steps_int <= 90:
payload["steps"] = steps_int
else:
return None, {"error": "Steps must be between 1-90"}
except ValueError:
return None, {"error": "Steps must be a valid number"}
if negative_prompt and negative_prompt.strip():
payload["negative_prompt"] = negative_prompt.strip()
if guidance:
try:
guidance_float = float(guidance)
if 1 <= guidance_float <= 28:
payload["guidance"] = guidance_float
else:
return None, {"error": "Guidance must be between 1-28"}
except ValueError:
return None, {"error": "Guidance must be a valid number"}
if seed:
try:
seed_int = int(seed)
payload["seed"] = seed_int
except ValueError:
return None, {"error": "Seed must be a valid number"}
try:
r = requests.post(f"{BASE_URL}/ai/image_generation", headers=headers, json=payload)
r.raise_for_status()
# The API returns the image directly as binary data
if r.headers.get('content-type', '').startswith('image/'):
# Convert bytes to PIL Image for Gradio
# Create PIL Image from bytes
image = Image.open(io.BytesIO(r.content))
return image, {"success": True, "message": "Image generated successfully!"}
else:
# If not an image, try to parse as JSON for error
try:
error_data = r.json()
return None, {"error": "Image generation failed", "details": error_data}
except:
return None, {"error": "Unexpected response format"}
except requests.exceptions.RequestException as req_err:
return None, {"error": "Request failed", "message": str(req_err)}
except Exception as e:
return None, {"error": "Unexpected error", "message": str(e)}
# ----------------- Gradio UI ------------------
with gr.Blocks() as demo:
gr.Markdown("""
<div style='text-align: center; margin-bottom: 24px;'>
<h1 style='font-size:2.2em; margin-bottom: 0.2em;'>🧩 Image Generation</h1>
<p style='font-size:1.2em; margin-top: 0;'>Generate high-quality images from text prompts using advanced AI models.</p>
<p style='font-size:1em; margin-top: 0.5em;'>For more details and API usage, see the <a href='https://jigsawstack.com/docs/api-reference/ai/image-generation' target='_blank'>documentation</a>.</p>
</div>
""")
with gr.Row():
with gr.Column():
gr.Markdown("#### Input")
prompt = gr.Textbox(
label="Text Prompt",
lines=4,
placeholder="Enter text to describe the image"
)
gr.Markdown("#### Quick Examples")
gr.Markdown("Click any example below to auto-fill the prompt:")
with gr.Row():
img_example_btn1 = gr.Button("πŸ”οΈ Mountain Landscape", size="sm")
img_example_btn2 = gr.Button("🐱 Cute Cat Portrait", size="sm")
img_example_btn3 = gr.Button("πŸ™οΈ Cyberpunk City", size="sm")
with gr.Row():
img_example_btn4 = gr.Button("🌊 Ocean Sunset", size="sm")
img_example_btn5 = gr.Button("πŸš€ Space Station", size="sm")
img_example_btn6 = gr.Button("🏰 Medieval Castle", size="sm")
with gr.Row():
img_example_btn7 = gr.Button("🎨 Abstract Art", size="sm")
img_example_btn8 = gr.Button("πŸ• Delicious Pizza", size="sm")
gr.Markdown("#### Image Options")
aspect_ratio = gr.Textbox(
label="Aspect Ratio (optional, e.g., 16:9, 4:3)",
placeholder="16:9, 4:3, etc.",
info="Common ratios: 16:9, 4:3, 1:1"
)
width = gr.Textbox(
label="Width (optional, pixels)",
placeholder="1024",
info="Minimum: 256, Maximum: 1920"
)
height = gr.Textbox(
label="Height (optional, pixels)",
placeholder="768",
info="Minimum: 256, Maximum: 1920"
)
steps = gr.Textbox(
label="Steps (optional, 1-90)",
placeholder="50",
info="Minimum: 1, Maximum: 90"
)
negative_prompt = gr.Textbox(
label="Negative Prompt (optional)",
placeholder="A bad image",
info="Optional: describe what you don't want in the image"
)
guidance = gr.Textbox(
label="Guidance (optional, 1-28)",
placeholder="7.5",
info="Minimum: 1, Maximum: 28"
)
seed = gr.Textbox(
label="Seed (optional)",
placeholder="Random",
info="Optional: randomize image generation"
)
with gr.Column():
gr.Markdown("#### Preview")
image_preview = gr.Image(label="Generated Image")
generate_image_btn = gr.Button("Generate Image", variant="primary")
generate_image_result = gr.JSON(label="Image Generation Result")
# Example functions to auto-fill prompt field
def fill_img_example_1():
return "A majestic mountain landscape with snow-capped peaks, crystal clear lake in the foreground, golden hour lighting, photorealistic, 8K resolution"
def fill_img_example_2():
return "A cute fluffy cat sitting on a windowsill, looking directly at camera with big green eyes, soft natural lighting, high quality, detailed fur"
def fill_img_example_3():
return "A futuristic cyberpunk cityscape at night, neon lights, flying cars, tall skyscrapers, rain-slicked streets, cinematic lighting"
def fill_img_example_4():
return "A breathtaking ocean sunset over calm waters, orange and pink sky, silhouettes of palm trees, peaceful atmosphere, high resolution"
def fill_img_example_5():
return "A massive space station orbiting Earth, detailed engineering, solar panels, astronauts in space suits, stars and nebula background"
def fill_img_example_6():
return "An ancient medieval castle on a hilltop, stone walls and towers, flags waving, dramatic clouds, fantasy atmosphere, detailed architecture"
def fill_img_example_7():
return "Abstract digital art with flowing colors, geometric shapes, vibrant gradients, modern aesthetic, artistic composition"
def fill_img_example_8():
return "A delicious pizza with melted cheese, pepperoni, fresh basil, wood-fired oven, appetizing, high quality food photography"
# Connect example buttons to auto-fill functions
img_example_btn1.click(fill_img_example_1, outputs=[prompt])
img_example_btn2.click(fill_img_example_2, outputs=[prompt])
img_example_btn3.click(fill_img_example_3, outputs=[prompt])
img_example_btn4.click(fill_img_example_4, outputs=[prompt])
img_example_btn5.click(fill_img_example_5, outputs=[prompt])
img_example_btn6.click(fill_img_example_6, outputs=[prompt])
img_example_btn7.click(fill_img_example_7, outputs=[prompt])
img_example_btn8.click(fill_img_example_8, outputs=[prompt])
generate_image_btn.click(
generate_image,
inputs=[prompt, aspect_ratio, width, height, steps, negative_prompt, guidance, seed],
outputs=[image_preview, generate_image_result]
)
demo.launch()