Update app.py

app.py

@@ -1,6 +1,11 @@
 from prodiapy import Prodia
 import gradio as gr
 import os
+import time
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.DEBUG)
 
 def generate_image(prompt):
     api_key = os.getenv("YOUR_PRODIA_KEY")
@@ -8,9 +13,20 @@ def generate_image(prompt):
         raise ValueError("API key not found in environment variables")
     
     prodia = Prodia(api_key=api_key)
-    job = prodia.sd.generate(prompt=prompt)
-    result = prodia.wait(job)
-    return result.image_url
+    
+    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")
 
 with gr.Blocks() as demo:
     gr.Markdown("## Prodia Stable Diffusion API")