Update app.py

app.py

@@ -1,15 +1,28 @@
-import prodiapy.aio
-import gradio as gr
+import os
 import asyncio
 
-async def generate(prompt):
-    prodia = prodiapy.aio.Prodia(api_key="your_api_key")
-    job = await prodia.sd.generate(prompt=prompt)
-    result = await prodia.wait(job)
+import prodiapy
+import gradio as gr
+
+# Load API key from Hugging Face Secrets
+api_key = os.getenv("PRODIA_AI_API")  
+prodia = prodiapy.Prodia(api_key=api_key)
+
+async def generate_image(prompt):
+    job = await prodia.sd.async_generate(prompt=prompt)
+    result = await prodia.async_wait(job)
     return result.image_url
 
-def wrapper(prompt):
-    return asyncio.run(generate(prompt))
+async def process_image(prompt):
+    return await asyncio.to_thread(generate_image, prompt) # Run in a thread
+
+demo = gr.Interface(
+    fn=process_image,
+    inputs="text",
+    outputs="image",
+    title="Async Prodia Image Generation",
+    description="Generate images asynchronously using Prodia API.",
+)
 
-demo = gr.Interface(fn=wrapper, inputs="text", outputs="image", title="Async Stable Diffusion", description="Generate images using a prompt asynchronously.")
-demo.launch()
+demo.queue() # Enable queuing to handle multiple requests
+demo.launch()