Update app.py

app.py

@@ -102,13 +102,22 @@ 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'
 
-    # Process the response from the model
-    response = ""
-    if response_data:
-        for result in response_data:
-            response += result.get('text', '')  # Process based on the response format
-
-    return response
+    # Check if the response is in the correct format (e.g., image)
+    try:
+        # Assuming the response is an image in base64 format
+        if '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
+        else:
+            return "Error: No valid image returned from the model."
+    except Exception as e:
+        return f"Error processing image: {e}"
 
 # Create the Gradio interface with an image input
 demo = gr.ChatInterface(