Update app.py

app.py

@@ -102,20 +102,20 @@ def respond(
     # Pass the base64-encoded image as the input
     response_data = client.text_to_image(images=image_base64, prompt=message)  # Pass the base64 string as 'images'
 
-    # Check if the response is in the correct format (e.g., image)
+    # Debug: print the response data to check its content
+    print("Response Data: ", response_data)
+    
     try:
-        # Assuming the response is an image in base64 format
-        if 'image' in response_data:
+        # Assuming the response is in the correct format, check for image content
+        if isinstance(response_data, dict) and 'image' in response_data:
             image_response = response_data['image']
-            print("Image Res:")
-            print(image_response)
             # Decode the base64 image back into an image object
             image_bytes = base64.b64decode(image_response)
             image = Image.open(io.BytesIO(image_bytes))
             image.show()  # Or return the image in Gradio
-            return "Image processed successfully"  # You can return some confirmation or process the image further
+            return "Image processed successfully"
         else:
-            return "Error: No valid image returned from the model."
+            return f"Error: Unexpected response format or no image data found. Response: {response_data}"
     except Exception as e:
         return f"Error processing image: {e}"