reverted to original

app.py

@@ -38,22 +38,20 @@ def call_api_and_generate_image(negative_prompt, positive_prompt,seed, source_im
     )
     
     result = resp.json()
-    base64_initial = result['result'][1]['data']
-    image_data = base64.b64decode(base64_initial)
-    image_output = Image.open(io.BytesIO(image_data))
-    
-    return image_output
+    base64_data = result['result'][0]['data']
+    image_data = base64.b64decode(base64_data)
+    image = Image.open(io.BytesIO(image_data))
+    return image, json.dumps(result, indent=2)
 
 def generate_image(positive_prompt, negative_prompt,source_image):
     seed = numpy.random.randint(0, 2**32 - 1)
     try:
-        image_output =  call_api_and_generate_image(negative_prompt, positive_prompt,seed,source_image)
-        return image_output,""
+        return call_api_and_generate_image(negative_prompt, positive_prompt,seed,source_image)
     except Exception as e:
         return None, f"Error: {str(e)}"
 
 def clear_fields():
-    return "", "", None  # Clear prompt and outputs
+    return "", "", None,""  # Clear prompt and outputs
 
 with gr.Blocks(theme='freddyaboulton/test-blue') as demo:
     gr.Markdown("<center><h2>Arjun's Image Generator</h2></center>")
@@ -67,8 +65,8 @@ with gr.Blocks(theme='freddyaboulton/test-blue') as demo:
                 clear_btn = gr.ClearButton(value="Clear Question and AI Response", elem_id="clear_button", variant="secondary", size="sm")
         gr.Markdown("<center><h3>AI Response</h3></center>")
         image_output_box = gr.Image(type="pil", label="Final Generated Image")
-
-    submit_btn.click(fn=generate_image, inputs = [prompt,neg_prompt,source_image], outputs=[image_output_box])
-    clear_btn.click(fn=clear_fields,outputs=[prompt,neg_prompt,image_output_box])
+        json_output = gr.Textbox(label="API Response JSON", lines=10)
+    submit_btn.click(fn=generate_image, inputs = [prompt,neg_prompt,source_image], outputs=[image_output_box,json_output])
+    clear_btn.click(fn=clear_fields,outputs=[prompt,neg_prompt,image_output_box,json_output])
 
 demo.launch()