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

# Final Settings
API_TOKEN = "Hf_xkdcoUnbTQbqSKWGLUwuHKFgFnhHcnrfYH"
API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
headers = {"Authorization": f"Bearer {API_TOKEN}"}

def generate_image(prompt):
    if not prompt:
        return None
    try:
        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
        if response.status_code == 200:
            return Image.open(io.BytesIO(response.content))
        else:
            # Ye line aapko batayegi ke error kya hai
            return f"Error: {response.status_code} - Model busy hai, 1 minute baad try karein."
    except Exception as e:
        return str(e)

with gr.Blocks() as demo:
    gr.Markdown("# 🛡️ NOMAN AI - POWER TEST")
    with gr.Row():
        input_t = gr.Textbox(label="Prompt")
        output_i = gr.Image(label="Result")
    btn = gr.Button("Banao Tasveer!")
    btn.click(fn=generate_image, inputs=input_t, outputs=output_i)

demo.launch()