Update app.py

app.py

@@ -1,8 +1,13 @@
 from prodiapy import Prodia
 import gradio as gr
+import os
 
 def generate_image(prompt):
-    prodia = Prodia(api_key="YOUR_PRODIA_KEY")
+    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)
     job = prodia.sd.generate(prompt=prompt)
     result = prodia.wait(job)
     return result.image_url