update code

app.py

@@ -184,10 +184,14 @@ def generate(state, imagebox, textbox, image_process_mode, gen_image, temperatur
         state.messages[-1][-1] = generated_text
         yield (state, state.to_gradio_chatbot(), "", None) + (disable_btn, disable_btn, disable_btn, enable_btn, enable_btn)
     
+    if gen_image == 'Yes':
+        print(generated_text)
+
     if gen_image == 'Yes' and '<h>' in generated_text and '</h>' in generated_text:
         common_neg_prompt = "out of frame, lowres, text, error, cropped, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, out of frame, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, username, watermark, signature"
         prompt = generated_text.split("<h>")[1].split("</h>")[0]
         generated_text = generated_text.split("<h>")[0] + '\n' + 'Prompt: ' + prompt + '\n'
+        print(prompt, '---------')
 
         torch.cuda.empty_cache()
         output_img = diffusion_pipe(prompt, negative_prompt=common_neg_prompt).images[0]
@@ -215,6 +219,7 @@ def add_text(state, imagebox, textbox, image_process_mode, gen_image):
         gen_image = 'Yes'
     elif 'show me one idea of what i could make with this?' in textbox.lower() and imagebox is not None:
         h, w = image.size
+        print(h, w, '=====')
         if h == 922 and w == 672:
             gen_image = 'Yes'
     if gen_image == 'Yes':