Update app.py

app.py

@@ -67,13 +67,14 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
     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:
@@ -88,7 +89,9 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
                 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:
@@ -105,18 +108,25 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
              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() # Keep traceback for unexpected errors
             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 ---
     final_prompt = f"{translated_prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
@@ -146,6 +156,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
              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>"
 
@@ -155,6 +167,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
              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>"
 
@@ -170,6 +184,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
                  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>"
 
@@ -189,6 +205,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
             # *** SUCCESS LOG ***
             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
 
@@ -196,6 +214,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
             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()
@@ -204,6 +224,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
             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>"
@@ -230,6 +252,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
         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 remain the same
@@ -247,6 +271,8 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
         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. Please check logs.</p>"