Update app.py

app.py

@@ -63,13 +63,14 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
     original_prompt = prompt
 
     # Translation
+    translated_prompt = prompt # Start with original
     try:
         translated_prompt = GoogleTranslator(source='auto', target='en').translate(prompt)
     except Exception as e:
         print(f"WARNING: Translation failed. Using original prompt.")
         print(f"  Error: {e}")
         print(f"  Prompt: '{original_prompt}'\n") # Add newline
-        translated_prompt = prompt
+        translated_prompt = prompt # Ensure it's the original if translation fails
 
     # --- OpenAI Moderation Check ---
     if openai_client:
@@ -84,7 +85,9 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
                 print("BLOCKED:")
                 print(f"  Reason: sexual/minors")
                 print(f"  Prompt: '{original_prompt}'")
-                print(f"  Translated: '{translated_prompt}'\n") # Add newline
+                if translated_prompt != original_prompt:
+                    print(f"  Translated: '{translated_prompt}'")
+                print("") # Add newline
                 return None, "<p style='color: red; text-align: center;'>Prompt violates safety guidelines. Generation blocked.</p>"
 
         except AuthenticationError:
@@ -101,18 +104,25 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
              print("BLOCKED:")
              print(f"  Reason: OpenAI Connection Error")
              print(f"  Prompt: '{original_prompt}'")
+             if translated_prompt != original_prompt:
+                 print(f"  Translated: '{translated_prompt}'")
              print(f"  Error: {e}\n") # Add newline
              return None, "<p style='color: red; text-align: center;'>Safety check failed. Please try again later.</p>"
         except Exception as e:
             print("BLOCKED:")
             print(f"  Reason: OpenAI Unexpected Error")
             print(f"  Prompt: '{original_prompt}'")
+            if translated_prompt != original_prompt:
+                print(f"  Translated: '{translated_prompt}'")
             print(f"  Error: {e}\n") # Add newline
             traceback.print_exc()
             return None, "<p style='color: red; text-align: center;'>An unexpected error occurred during safety check. Generation blocked.</p>"
     else:
         print(f"WARNING: OpenAI client not available. Skipping moderation.")
-        print(f"  Prompt: '{original_prompt}'\n") # Add newline
+        print(f"  Prompt: '{original_prompt}'") # Log original prompt even if skipping
+        if translated_prompt != original_prompt:
+             print(f"  (Would use translated: '{translated_prompt}')")
+        print("") # Add newline
 
     # --- Proceed with Generation ---
     # Add suffix (Ensure consistency with gradio1.py if desired, or keep model-specific suffixes)
@@ -148,6 +158,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
              print("FAILED:")
              print(f"  Reason: Invalid Image Data (Empty/Small)")
              print(f"  Prompt: '{original_prompt}'")
+             if translated_prompt != original_prompt:
+                 print(f"  Translated: '{translated_prompt}'")
              print(f"  Length: {len(image_bytes)}\n") # Add newline
              return None, "<p style='color: red; text-align: center;'>API returned invalid image data.</p>"
 
@@ -157,6 +169,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
              print("FAILED:")
              print(f"  Reason: Image Processing Error")
              print(f"  Prompt: '{original_prompt}'")
+             if translated_prompt != original_prompt:
+                 print(f"  Translated: '{translated_prompt}'")
              print(f"  Error: {img_err}\n") # Add newline
              return None, "<p style='color: red; text-align: center;'>Failed to process image data from API.</p>"
 
@@ -172,6 +186,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
                  print("FAILED:")
                  print(f"  Reason: Image Save Verification Error")
                  print(f"  Prompt: '{original_prompt}'")
+                 if translated_prompt != original_prompt:
+                      print(f"  Translated: '{translated_prompt}'")
                  print(f"  Path: '{save_path}'\n") # Add newline
                  return image, "<p style='color: red; text-align: center;'>Internal Error: Failed to confirm image file save.</p>"
 
@@ -192,6 +208,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
             # *** SUCCESS LOG *** (Updated format)
             print("SUCCESS:")
             print(f"  Prompt: '{original_prompt}'")
+            if translated_prompt != original_prompt:
+                 print(f"  Translated: '{translated_prompt}'")
             print(f"  {arinteli_url}\n") # Add newline
             return image, download_html
 
@@ -199,6 +217,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
             print("FAILED:")
             print(f"  Reason: Image Save IO Error")
             print(f"  Prompt: '{original_prompt}'")
+            if translated_prompt != original_prompt:
+                 print(f"  Translated: '{translated_prompt}'")
             print(f"  Path: '{save_path}'")
             print(f"  Error: {save_err}\n") # Add newline
             traceback.print_exc()
@@ -207,6 +227,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
             print("FAILED:")
             print(f"  Reason: Link Creation/Save Unexpected Error")
             print(f"  Prompt: '{original_prompt}'")
+            if translated_prompt != original_prompt:
+                 print(f"  Translated: '{translated_prompt}'")
             print(f"  Error: {e}\n") # Add newline
             traceback.print_exc()
             return image, "<p style='color: red; text-align: center;'>Internal Error creating download link.</p>"
@@ -233,6 +255,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
         print("FAILED:")
         print(f"  Reason: HF API HTTP Error {status_code}")
         print(f"  Prompt: '{original_prompt}'")
+        if translated_prompt != original_prompt:
+             print(f"  Translated: '{translated_prompt}'")
         print(f"  Details: '{error_text[:200]}'\n") # Add newline
 
         # User-facing messages (Keep consistent with previous version of this file)
@@ -249,6 +273,8 @@ def query(prompt, negative_prompt="", steps=4, cfg_scale=0, seed=-1, width=1024,
         print("FAILED:")
         print(f"  Reason: Unexpected Error During Generation")
         print(f"  Prompt: '{original_prompt}'")
+        if translated_prompt != original_prompt:
+             print(f"  Translated: '{translated_prompt}'")
         print(f"  Error: {e}\n") # Add newline
         traceback.print_exc()
         return None, f"<p style='color: red; text-align: center;'>An unexpected error occurred: {e}</p>" # Keep specific error for user
@@ -272,7 +298,6 @@ textarea:focus {
 
 # Build the Gradio UI with Blocks
 with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
-    gr.HTML("<center><h1>FLUX.1-Schnell</h1></center>")
 
     with gr.Column(elem_id="app-container"):
         with gr.Row():