Update app.py

app.py

@@ -18,7 +18,10 @@ def generate_comic(prompt):
         return f"Error: {response.status_code}, {response.text}"
     
     # Extracting generated images
-    images = response.json()["generated_images"]
+    images = response.json().get("generated_images", [])
+    
+    if not images:
+        return "No images were generated, please try again with a different prompt."
     
     # Convert image bytes into PIL Image format
     pil_images = [Image.open(io.BytesIO(img)) for img in images]
@@ -33,15 +36,15 @@ def gradio_interface():
         
         with gr.Row():
             prompt = gr.Textbox(label="Enter your short story description", placeholder="Once upon a time...")
-        
-        # Directly setting grid and height in Gallery constructor
-        output_gallery = gr.Gallery(label="Generated Comic Panels", elem_id="gallery", columns=3).style(height="300px")
-        
+
+        # Directly specifying the number of columns without .style()
+        output_gallery = gr.Gallery(label="Generated Comic Panels", columns=3, height=300)
+
         submit_button = gr.Button("Generate Comic")
-        
+
         # Set up button functionality
         submit_button.click(fn=generate_comic, inputs=prompt, outputs=output_gallery)
-    
+
     return demo
 
 # Run the Gradio app