One output

app.py

@@ -39,14 +39,10 @@ def call_api_and_generate_image(negative_prompt, positive_prompt,seed, source_im
     
     result = resp.json()
     base64_initial = result['result'][0]['data']
-    initial_image_data = base64.b64decode(base64_initial)
-    initial_image = Image.open(io.BytesIO(initial_image_data))
+    image_data = base64.b64decode(base64_initial)
+    image_output = Image.open(io.BytesIO(image_data))
     
-    base64_final = result['result'][1]['data']
-    final_image_data = base64.b64decode(base64_final)
-    final_image = Image.open(io.BytesIO(final_image_data))
-    
-    return initial_image,final_image
+    return image_output
 
 def generate_image(positive_prompt, negative_prompt,source_image):
     seed = numpy.random.randint(0, 2**32 - 1)
@@ -56,7 +52,7 @@ def generate_image(positive_prompt, negative_prompt,source_image):
         return None, f"Error: {str(e)}"
 
 def clear_fields():
-    return "", "", None, 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>")
@@ -69,10 +65,9 @@ with gr.Blocks(theme='freddyaboulton/test-blue') as demo:
                 submit_btn = gr.Button(value="Submit", elem_id="generate_button", variant="primary", size="sm")
                 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>")
-        initial_image_output = gr.Image(type="pil", label="Initial Generated Image")
-        final_image_output = gr.Image(type="pil", label="Final Image After Face Swap")
+        image_output = gr.Image(type="pil", label="Final Generated Image")
 
-    submit_btn.click(fn=generate_image, inputs = [prompt,neg_prompt,source_image], outputs=[initial_image_output,final_image_output])
-    clear_btn.click(fn=clear_fields,outputs=[prompt,neg_prompt,initial_image_output,final_image_output])
+    submit_btn.click(fn=generate_image, inputs = [prompt,neg_prompt,source_image], outputs=[image_output])
+    clear_btn.click(fn=clear_fields,outputs=[prompt,neg_prompt,image_output])
 
 demo.launch()