Update app.py

app.py

@@ -59,14 +59,13 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
     try:
         # Using 'auto' source detection
         translated_prompt = GoogleTranslator(source='auto', target='en').translate(prompt)
-        print(f'Generation {key} translation: {translated_prompt}')
     except Exception as e:
         print(f"Translation failed: {e}. Using original prompt.")
         translated_prompt = prompt # Fallback to original if translation fails
 
     # Add suffix to prompt
     final_prompt = f"{translated_prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
-    print(f'Generation {key} final prompt: {final_prompt}')
+    print(f'Generation {key} prompt: {final_prompt}')
 
     # Prepare payload for API call
     payload = {
@@ -84,7 +83,6 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
 
     # API Call Section
     try:
-        print(f"Sending request to API: {API_URL}")
         if not headers:
              print("WARNING: Authorization header is missing (HF_READ_TOKEN not set?)")
              # Handle error appropriately - maybe return an error message
@@ -101,7 +99,6 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
 
         try:
             image = Image.open(io.BytesIO(image_bytes))
-            print(f"Image received and opened successfully. Format: {image.format}, Size: {image.size}")
         except UnidentifiedImageError as img_err:
              print(f"Error: Could not identify or open image from API response bytes: {img_err}")
              # Optionally save the invalid bytes for debugging
@@ -110,8 +107,6 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
              # print(f"Saved problematic bytes to {error_bytes_path}")
              return None, "<p style='color: red; text-align: center;'>Failed to process image data from API.</p>"
 
-        print(f'Generation {key} API call successful!')
-
         # --- Save image and create download link ---
         filename = f"{int(time.time())}_{uuid.uuid4().hex[:8]}.png"
         # save_path is relative to the script's execution directory
@@ -119,15 +114,12 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
         absolute_save_path = os.path.abspath(save_path) # Get absolute path for logging
 
         try:
-            print(f"Attempting to save image to: {absolute_save_path}")
             # Save image explicitly as PNG
             image.save(save_path, "PNG")
 
             # *** Verify file exists after saving ***
             if os.path.exists(save_path):
                 file_size = os.path.getsize(save_path)
-                print(f"SUCCESS: Image confirmed saved to: {save_path} (Absolute: {absolute_save_path})")
-                print(f"Saved file size: {file_size} bytes")
                 if file_size < 100: # Warn if the saved file is suspiciously small
                      print(f"WARNING: Saved file {save_path} is very small ({file_size} bytes). May indicate an issue.")
                      # Optionally return a warning message in the UI
@@ -139,15 +131,13 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
 
             # Get current space name from the API URL
             space_name = "greendra-stable-diffusion-3-5-large-serverless"
-            print(f"Current space name: {space_name}")
             
             relative_file_url = f"/gradio_api/file={save_path}"
-            print(f"Generated relative file URL for Gradio API: {relative_file_url}")
 
             encoded_file_url = quote(relative_file_url)
             # Add space_name parameter to the URL
             arintelli_url = f"{ARINTELLI_REDIRECT_BASE}?download_url={encoded_file_url}&space_name={space_name}"
-            print(f"Generated redirect link: {arintelli_url}")
+            print(f"{arintelli_url}")
 
             # Use simpler button style like the Run button
             download_html = (
@@ -158,7 +148,7 @@ def query(prompt, negative_prompt, steps=35, cfg_scale=7, seed=-1, width=1024, h
                 f'</div>'
             )
 
-            print(f"--- Generation {key} Completed Successfully ---")
+            print(f"--- Generation {key} Done ---")
             return image, download_html
 
         except (OSError, IOError) as save_err: