Spaces:
Running
Running
File size: 3,948 Bytes
6142acc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import requests
import gradio as gr
import os
import json
import base64
# 📝 Form-like Configuration Section - Easily customize your script settings here!
# -----------------------------------------------------------
# Server Configuration - Change these to match your server!
fooocus_host = 'http://192.168.50.136:7878' # 🌐 Host address for the Fooocus server
# 📁 File Paths - Change these paths if your images are stored in a different location!
imgs_base_path = "./saved_images" # Define the base path for saving images
# Headers for HTTP Requests - Modify if needed!
headers = {
"accept": "application/json"
}
# -----------------------------------------------------------
# 🛠️ Utility Functions - These functions perform the core operations of the script.
# Function to encode images to base64 - This helps in preparing images for API requests.
def encode_image_to_base64(image_path):
with open(image_path, "rb") as image_file:
return base64.b64encode(image_file.read()).decode('utf-8')
# Function to save images from API response - This saves the processed images locally.
def save_image_from_response(response, base_filename="generated-image"):
if not os.path.exists(imgs_base_path):
os.makedirs(imgs_base_path) # Create the directory if it does not exist
output_path = os.path.join(imgs_base_path, base_filename)
for i, item in enumerate(response, start=1):
if item.get('url'):
modified_url = item['url'].replace('http://127.0.0.1:7878', fooocus_host)
img_data = requests.get(modified_url).content
with open(f"{output_path}{i}.png", 'wb') as handler:
handler.write(img_data)
print(f"🖼️ Image saved as {output_path}{i}.png")
# Enhanced Gradio app for text and image prompt generation
def generate_image_from_prompt(prompt, image_file=None):
api_url = f"{fooocus_host}/v1/generation/text-to-image"
request_payload = {"prompt": prompt}
if image_file:
image_base64 = encode_image_to_base64(image_file.name)
request_payload["image_prompt"] = image_base64 # Add image prompt if provided
response = requests.post(api_url, json=request_payload, headers=headers)
if response.status_code == 200:
result = response.json()
if isinstance(result, list) and len(result) > 0:
save_image_from_response(result, "generated-image")
image_url = result[0].get("url") # Assuming the first item in the list contains the URL
if image_url:
modified_url = image_url.replace('http://127.0.0.1:7878', fooocus_host)
return f"<img src='{modified_url}'/>" # Return an HTML image tag with the modified URL
else:
return "Error: URL not found in response."
elif isinstance(result, dict):
save_image_from_response([result], "generated-image-single") # Save single image response
image_url = result.get("url")
if image_url:
modified_url = image_url.replace('http://127.0.0.1:7878', fooocus_host)
return f"<img src='{modified_url}'/>" # Return an HTML image tag with the modified URL
else:
return "Error: URL not found in response."
else:
return "Error: Unexpected response format. Please try again."
else:
return f"Error: Could not generate image. Status code: {response.status_code}"
iface = gr.Interface(
fn=generate_image_from_prompt,
inputs=[gr.Textbox(label="Enter your prompt"), gr.File(label="Upload an image prompt (optional)")],
outputs=gr.HTML(label="Generated Image"), # Adjusted to handle image URLs via HTML
title="Enhanced Image Generator",
description="Enter a prompt and optionally upload an image to generate an image using the API. Images are saved locally."
)
if __name__ == "__main__":
iface.launch()
|