| import gradio as gr |
| import requests |
| import json |
| import time |
|
|
| url = "https://nexra.aryahcr.cc/api/image/complements" |
| headers = { |
| "Content-Type": "application/json" |
| } |
|
|
| def generate_image(prompt, model): |
| data = { |
| "prompt": prompt, |
| "model": model, |
| } |
| try: |
| response = requests.post(url, headers=headers, data=json.dumps(data)) |
| |
| if response.status_code == 200: |
| response_data = response.json() |
| image_id = response_data.get("id") |
| |
| if not image_id: |
| return "Error: No image ID returned in the response." |
|
|
| while True: |
| status_response = requests.get(f"{url}/{image_id}") |
| |
| if status_response.status_code == 200: |
| status_data = status_response.json() |
| status = status_data.get("status") |
| |
| if status == "completed": |
| images = status_data.get("images") |
| if images and isinstance(images, list): |
| image_url = images[0] |
| return image_url |
| else: |
| return "Error: No images found in the response." |
| elif status == "error": |
| return "Error: Image generation failed." |
| elif status == "not_found": |
| return "Error: Image ID not found." |
| elif status == "pending": |
| time.sleep(1) |
| else: |
| return f"Error: Unexpected status '{status}'." |
| else: |
| return f"Error: Status check failed with code {status_response.status_code}." |
| else: |
| return f"Error: Initial request failed with code {response.status_code}." |
| except json.JSONDecodeError: |
| return "Error: Invalid JSON response." |
| except Exception as e: |
| return f"Exception occurred: {str(e)}" |
|
|
| iface = gr.Interface( |
| fn=generate_image, |
| inputs=[ |
| gr.Textbox(label="Enter prompt"), |
| gr.Radio(["dalle2"], label="Select Model") |
| ], |
| outputs=gr.Image(type="filepath"), |
| title="DALLE2 Generation", |
| description="DALLE-2 Generation from Nextra API, NOTE: DALL-E 3 is not supported yet." |
| ) |
| iface.launch() |
