Update app.py

app.py

@@ -48,7 +48,7 @@ class Prodia:
         response = requests.post(url, headers=headers, data=json.dumps(params))
 
         if response.status_code != 200:
-            raise Exception(f"Bad Prodia Response: {response.status_code}")
+            raise Exception(f"Bad Prodia Response: {response.status_code} - {response.text}")
 
         return response
 
@@ -56,42 +56,47 @@ class Prodia:
         response = requests.get(url, headers=self.headers)
 
         if response.status_code != 200:
-            raise Exception(f"Bad Prodia Response: {response.status_code}")
+            raise Exception(f"Bad Prodia Response: {response.status_code} - {response.text}")
 
         return response
 
-
 def image_to_base64(image_path):
-    # Open the image with PIL
     with Image.open(image_path) as image:
-        # Convert the image to bytes
         buffered = BytesIO()
-        image.save(buffered, format="PNG")  # You can change format to PNG if needed
-
-        # Encode the bytes to base64
+        image.save(buffered, format="PNG")
         img_str = base64.b64encode(buffered.getvalue())
+    return img_str.decode('utf-8')
 
-    return img_str.decode('utf-8')  # Convert bytes to string
+# Ensure that the API key is available
+api_key = os.getenv("PRODIA_API_KEY")
+if not api_key:
+    raise ValueError("Prodia API key not found in environment variables")
 
-
-prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
+prodia_client = Prodia(api_key=api_key)
 
 def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
-    result = prodia_client.generate({
-        "prompt": prompt,
-        "negative_prompt": negative_prompt,
-        "model": model,
-        "steps": steps,
-        "sampler": sampler,
-        "cfg_scale": cfg_scale,
-        "width": width,
-        "height": height,
-        "seed": seed
-    })
-
-    job = prodia_client.wait(result)
-
-    return job["imageUrl"]
+    try:
+        result = prodia_client.generate({
+            "prompt": prompt,
+            "negative_prompt": negative_prompt,
+            "model": model,
+            "steps": steps,
+            "sampler": sampler,
+            "cfg_scale": cfg_scale,
+            "width": width,
+            "height": height,
+            "seed": seed
+        })
+
+        job = prodia_client.wait(result)
+
+        if job['status'] == 'succeeded':
+            return job["imageUrl"]
+        else:
+            return "Generation failed, please try again."
+
+    except Exception as e:
+        return f"An error occurred: {e}"
 
 css = """
 /* Overall Styling */
@@ -132,7 +137,7 @@ body {
 
 /* Button Styling */
 #generate {
-    background-color: #007bff; /* Example - use your preferred color */
+    background-color: #007bff; 
     color: white;
     padding: 15px 25px;
     border: none;
@@ -141,10 +146,10 @@ body {
 }
 
 #generate:hover {
-    background-color: #0056b3; /* Darker shade on hover */
+    background-color: #0056b3; 
 }
 
-/* Responsive Design - Adjust breakpoints as needed */
+/* Responsive Design */
 @media screen and (max-width: 768px) {
     #settings {
         grid-template-columns: 1fr; 
@@ -152,10 +157,8 @@ body {
 }
 """
 
-# --- Gradio Interface ---
-
 with gr.Blocks(css=css) as demo:
-    state = gr.State(value="Welcome Screen")  # To control the visibility of tabs
+    state = gr.State(value="Welcome Screen")
 
     with gr.Tabs() as tabs:
         with gr.TabItem("Welcome Screen"):