attempt 4

app_v3.py

@@ -70,7 +70,7 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
             control_guidance_start=0.0,
             control_guidance_end=guidance_end,
         ).images[0]
-
+        print("Type: " + str(type(image)))
     return image
 
 def combine_caption_focus(caption, focus):
@@ -108,10 +108,11 @@ def process_image(control_image, user_prompt, system_prompt, scale, steps,
                 guidance_end, temperature, top_p, max_new_tokens, log_prompt):
     # Initialize with empty caption
     final_prompt = user_prompt.strip()
-    
+    print("User generated prompt: " + final_prompt)
     # If no user prompt provided, generate a caption first
     if not final_prompt:
         # Generate a detailed caption
+        print("Generating caption...")
         mcaption = model.caption(control_image, length="normal")
         detailed_caption = mcaption["caption"]
         final_prompt = detailed_caption
@@ -135,15 +136,9 @@ def process_image(control_image, user_prompt, system_prompt, scale, steps,
         log_params(final_prompt, scale, steps, controlnet_conditioning_scale, guidance_scale, seed, guidance_end, control_image, image)
         yield f"Completed! Used prompt: {final_prompt}", image, final_prompt
     except Exception as e:
+        print("Error: " + str(e))
         yield f"Error: {str(e)}", None, None
     
-def handle_outputs(outputs):
-    if isinstance(outputs, dict) and outputs.get("__type__") == "update_caption":
-        return outputs["caption"], None
-    return outputs
-
-
-
 with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
     gr.Markdown("⚠️ WIP SPACE - UNFINISHED & BUGGY")
     with gr.Row():