Update app.py

app.py

@@ -48,27 +48,29 @@ def generate_text(prompt, temperature=0.9, max_new_tokens=512, top_p=0.95, repet
 def generate_image(prompt):
     try:
         result = image_client.predict(
-            prompt=prompt,
-            negative_prompt="",
-            seed=0,
-            custom_width=1024,
-            custom_height=1024,
-            guidance_scale=7.0,
-            num_inference_steps=28,
-            sampler="Euler a",
-            aspect_ratio_selector="896 x 1152",
-            style_selector="(None)",
-            quality_selector="Standard v3.1",
-            use_upscaler=False,
-            upscaler_strength=0.55,
-            upscale_by=1.5,
-            add_quality_tags=True,
-            isImg2Img=False,
-            img_path=None,
-            img2img_strength=0.65,
+            prompt,  # Image prompt
+            "",  # Negative prompt
+            0,  # Seed
+            1024,  # Width
+            1024,  # Height
+            7.0,  # Guidance scale
+            28,  # Number of inference steps
+            'Euler a',  # Sampler
+            '896 x 1152',  # Aspect Ratio
+            '(None)',  # Style Preset
+            'Standard v3.1',  # Quality Tags Presets
+            False,  # Use Upscaler
+            0.55,  # Upscaler strength
+            1.5,  # Upscale by
+            True,  # Add Quality Tags
             api_name="/run"
         )
-        return result[0]['image']
+        # Check if result is not empty and contains the expected structure
+        if result and isinstance(result, list) and len(result) > 0 and 'image' in result[0]:
+            return result[0]['image']
+        else:
+            st.error("Unexpected result format from the Gradio API.")
+            return None
     except Exception as e:
         st.error(f"Error generating image: {e}")
         st.write("Full error details:", e)