Update app.py

app.py

@@ -7,6 +7,23 @@ from diffusers import  DiffusionPipeline, FlowMatchEulerDiscreteScheduler, Autoe
 from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 
+import logging
+import sys
+from datetime import datetime
+import gc
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.StreamHandler(sys.stdout),
+        logging.FileHandler('transcription.log')
+    ]
+)
+logger = logging.getLogger(__name__)
+
+
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -74,6 +91,8 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             output_type="pil",
             good_vae=good_vae,
         ):
+            logger.info(f"PROMPT: {prompt}")
+        
             yield img, seed
     
 examples = [
@@ -169,7 +188,7 @@ with gr.Blocks(css=css) as demo:
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
-        fn = [infer, infer_conditional, infer_conditional_test],
+        fn = infer,
         inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result, seed]
     )