import gradio as gr
import requests
from PIL import Image
from io import BytesIO

print("✅ VoiceForge API Server Booting up! (Dev: Rahul Terpathi)")

def generate_image(prompt):
    if not prompt: return None
    try:
        # Background mein unlimited Pollinations AI ko hit karega
        safe_prompt = prompt.replace(" ", "%20")
        url = f"https://image.pollinations.ai/prompt/{safe_prompt}?width=1024&height=1024&nologo=true"
        
        response = requests.get(url)
        image = Image.open(BytesIO(response.content))
        return image
    except Exception as e:
        return None

# Clean Dark Mode UI (Sirf API ke liye)
with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
    gr.Markdown("# ⚙️ VoiceForge Background Engine")
    
    with gr.Row():
        inp = gr.Textbox(label="Input Prompt")
        btn = gr.Button("Generate")
    out = gr.Image(label="Output Image")
    
    # Ye api_name="generate" hi tere Netlify wale frontend se connect hoga
    btn.click(fn=generate_image, inputs=inp, outputs=out, api_name="generate")

demo.launch()