gradio app v4.1

app.py

@@ -1,7 +1,85 @@
 import gradio as gr
+import aiohttp
+import asyncio
+import random
+import io
+from PIL import Image
+import os
+from concurrent.futures import ThreadPoolExecutor
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+markdown_link = "For more generations, visit [Imagine Art](https://www.imagine.art/)."
+
+TOKEN = os.getenv("TOKEN")
+URL = os.getenv("URL")
+print(f"URL: {URL}, Type: {type(URL)}")
+
+if not URL:
+    raise ValueError("URL environment variable is not set.")
+
+async def async_generation_request(request_data, is_user_premium):
+    seed = request_data.get("seed", random.randint(1, 10000000))
+
+    headers = {
+        "bearer": TOKEN,
+        "Style-Id": request_data["model_id"]
+    }
+    if is_user_premium:
+        headers["Premium"] = "True"
+
+    for key, value in headers.items():
+        if value is None:
+            raise ValueError(f"Header value is missing for key: {key}")
+
+    async with aiohttp.ClientSession() as session:
+        for i in range(3):
+            data = aiohttp.FormData()
+            data.add_field('prompt', request_data["prompt"], content_type='multipart/form-data')
+            data.add_field('style_id', request_data["model_id"])
+            data.add_field('seed', str(seed))
+            data.add_field('aspect_ratio', request_data["aspect_ratio"])
+            data.add_field('negative_prompt', request_data["negative_prompt"])
+            data.add_field('cfg', str(request_data["cfg_scale"]))
+
+            async with session.post(URL, data=data, headers=headers) as response:
+                if response.status == 200:
+                    return await response.read()
+                print(f"Failed API Call: {response.status} >>> {await response.text()}")
+
+    return None
+
+def run_async_in_thread(func, *args):
+    with ThreadPoolExecutor() as executor:
+        return executor.submit(asyncio.run, func(*args)).result()
+
+def gradio_generation_request(prompt, aspect_ratio, negative_prompt, cfg_scale):
+    request_data = {
+        "prompt": prompt,
+        "model_id": "32",
+        "aspect_ratio": aspect_ratio,
+        "negative_prompt": negative_prompt,
+        "cfg_scale": cfg_scale,
+        "seed": random.randint(1, 10000000)
+    }
+
+    output = run_async_in_thread(async_generation_request, request_data, True)  
+
+    if output:
+        return Image.open(io.BytesIO(output))
+    else:
+        return None
+
+interface = gr.Interface(
+    fn=gradio_generation_request,
+    inputs=[
+        gr.Textbox(label="Prompt"),
+        gr.Textbox(label="Aspect Ratio"),
+        gr.Textbox(label="Negative Prompt"),
+        gr.Slider(minimum=1, maximum=15, value=7.5, label="CFG Scale") 
+    ],
+    outputs="image",
+    title='IMAGINE V4.1 GENERATION',
+    description=markdown_link
+)
+
+interface.launch()

flagged/log.csv

@@ -0,0 +1,3 @@
+Prompt,Model ID,Aspect Ratio,Negative Prompt,CFG Scale,Is User Premium,output,flag,username,timestamp
+girl eating banana,35,,,7.5,True,/home/evobits/rashid/imagineV4.1/flagged/output/723e768be4fee0202a2ee36bde863103fb3fbcda/tmpmxb31ukx.png,,,2024-01-09 19:26:31.119421
+,,,7.5,,,,2024-01-10 11:25:22.338663