Update app_v2.py

app_v2.py

@@ -79,6 +79,10 @@ def caption(input_image: Image.Image, prompt: str, temperature: float, top_p: fl
 
     output = cap_model.generate(**generate_kwargs)
     print(f"Generated {len(output[0])} tokens")
+    print(f"Generated {type(output)}")
+    print(f"Generated {output}")
+
+    #return output[0]
 
 @spaces.GPU()
 @torch.no_grad()
@@ -167,7 +171,7 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as iface:
         generated_image = gr.Image(type="pil", label="Generated Image", format="png", show_label=False)
     with gr.Row():
         with gr.Column(scale=1):
-            prompt = gr.Textbox(lines=4, placeholder="Enter your prompt here...", label="Prompt")
+            prompt = gr.Textbox(lines=4, placeholder="Enter your prompt here...", label="Prompt", interactive=True)
             output_caption = gr.Textbox(label="Caption")
             scale = gr.Slider(1, 3, value=1, label="Scale", step=0.25)
             generate_button = gr.Button("Generate Image", variant="primary")
@@ -184,7 +188,7 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as iface:
                 lines=4, 
                 value=sys_prompt,
                 label="System Prompt for Captioning",
-                visible=True  # Changed to visible
+                visible=False  # Changed to visible
             )
             temperature_slider = gr.Slider(
                 minimum=0.0, maximum=2.0, value=0.6, step=0.05,
@@ -214,7 +218,7 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as iface:
             controlnet_conditioning_scale, guidance_scale, seed, 
             guidance_end, temperature_slider, top_p_slider, max_tokens_slider, log_prompt
         ],
-        outputs=[output_caption, generated_image]
+        outputs=[prompt, generated_image]
     )
     
     caption_button.click(