Update app.py

app.py

@@ -1,32 +1,17 @@
-from prodiapy import Prodia
 import gradio as gr
 import os
-import time
-import logging
-
-# Configure logging
-logging.basicConfig(level=logging.DEBUG)
+from prodia import createProdia
 
 def generate_image(prompt):
     api_key = os.getenv("YOUR_PRODIA_KEY")
     if not api_key:
         raise ValueError("API key not found in environment variables")
     
-    prodia = Prodia(api_key=api_key)
+    prodia = createProdia({"apiKey": api_key})
     
-    attempt = 0
-    max_attempts = 3
-    while attempt < max_attempts:
-        try:
-            job = prodia.sd.generate(prompt=prompt)
-            result = prodia.wait(job)
-            return result.image_url
-        except Exception as e:
-            logging.error(f"Attempt {attempt+1} failed: {e}")
-            attempt += 1
-            time.sleep(2)  # wait before retrying
-
-    raise Exception("Failed to generate image after multiple attempts")
+    job = prodia.generate({"prompt": prompt})
+    result = prodia.wait(job)
+    return result["imageUrl"]
 
 with gr.Blocks() as demo:
     gr.Markdown("## Prodia Stable Diffusion API")