Please enter a prompt.
" key = random.randint(0, 999) print(f"\n--- Generation {key} Started ---") # Translation try: # Using 'auto' source detection translated_prompt = GoogleTranslator(source='auto', target='en').translate(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} prompt: {final_prompt}') # Prepare payload for API call payload = { "inputs": final_prompt, "parameters": { # Nested parameters as per original structure "width": width, "height": height, "num_inference_steps": steps, "negative_prompt": negative_prompt, "guidance_scale": cfg_scale, "seed": seed if seed != -1 else random.randint(1, 1000000000), } # Add other parameters here if needed (e.g., sampler if supported) } # API Call Section try: if not headers: print("WARNING: Authorization header is missing (HF_READ_TOKEN not set?)") # Handle error appropriately - maybe return an error message return None, "Configuration Error: API Token missing.
" response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout) response.raise_for_status() # Raises HTTPError for 4xx/5xx responses image_bytes = response.content # Check for valid image data before proceeding if not image_bytes or len(image_bytes) < 100: # Basic check for empty/tiny response print(f"Error: Received empty or very small response content (length: {len(image_bytes)}). Potential API issue.") return None, "API returned invalid image data.
" try: image = Image.open(io.BytesIO(image_bytes)) 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 # error_bytes_path = os.path.join(IMAGE_DIR, f"error_{key}_bytes.bin") # with open(error_bytes_path, "wb") as f: f.write(image_bytes) # print(f"Saved problematic bytes to {error_bytes_path}") return None, "Failed to process image data from API.
" # --- 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 save_path = os.path.join(IMAGE_DIR, filename) absolute_save_path = os.path.abspath(save_path) # Get absolute path for logging try: # 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) 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 # return image, "Warning: Saved image file is unexpectedly small.
" else: # This indicates a serious problem if save() didn't raise an error but the file isn't there print(f"CRITICAL ERROR: File NOT found at {save_path} (Absolute: {absolute_save_path}) immediately after saving!") return image, "Internal Error: Failed to confirm image file save.
" # Get current space name from the API URL space_name = "greendra-stable-diffusion-3-5-large-serverless" relative_file_url = f"/gradio_api/file={save_path}" encoded_file_url = quote(relative_file_url) # Add space_name parameter to the URL arinteli_url = f"{ARINTELI_REDIRECT_BASE}?download_url={encoded_file_url}&space_name={space_name}" print(f"{arinteli_url}") # Use simpler button style like the Run button download_html = ( f'Internal Error: Failed to save image file. Details: {save_err}
" except Exception as e: # Catch any other unexpected errors during link creation/saving print(f"Error during link creation or unexpected save issue: {e}") traceback.print_exc() # Return the generated image (if available) but indicate link error return image, "Internal Error creating download link.
" # --- Exception Handling for API Call --- except requests.exceptions.Timeout: print(f"Error: Request timed out after {timeout} seconds.") return None, "Request timed out. The model is taking too long.
" except requests.exceptions.HTTPError as e: # Handle HTTP errors from the API (4xx, 5xx) status_code = e.response.status_code error_text = e.response.text # Default error text try: # Try to parse more specific error message from JSON response error_data = e.response.json() error_text = error_data.get('error', error_text) if isinstance(error_text, dict) and 'message' in error_text: error_text = error_text['message'] # Handle nested messages except json.JSONDecodeError: pass # Keep raw text if not JSON print(f"Error: Failed API call. Status: {status_code}, Response: {error_text}") # Generate user-friendly messages based on status code if status_code == 503: # Service Unavailable (often model loading) estimated_time = error_data.get("estimated_time") if 'error_data' in locals() and isinstance(error_data, dict) else None if estimated_time: error_message = f"Model is loading (503), please wait. Est. time: {estimated_time:.1f}s. Try again." else: error_message = f"Service unavailable (503). Model might be loading or down. Try again later." elif status_code == 400: # Bad Request (invalid parameters) error_message = f"Bad Request (400): Check parameters. API Error: {error_text}" elif status_code == 422: # Unprocessable Entity (validation error) error_message = f"Validation Error (422): Input invalid. API Error: {error_text}" elif status_code == 401 or status_code == 403: # Unauthorized / Forbidden error_message = f"Authorization Error ({status_code}): Check your API Token (HF_READ_TOKEN)." else: # Generic API error error_message = f"API Error: {status_code}. Details: {error_text}" # Return None for image, and the error message string for the HTML component return None, f"{error_message}
" except Exception as e: # Catch any other unexpected errors during the process print(f"An unexpected error occurred: {e}") traceback.print_exc() # Log detailed traceback return None, f"An unexpected error occurred: {e}
" # --- CSS Styling --- css = """ #app-container { max-width: 800px; margin-left: auto; margin-right: auto; } textarea:focus { background: #0d1117 !important; } #download-link-container p { /* Style the link container */ margin-top: 10px; /* Add some space above the link */ font-size: 0.9em; /* Slightly smaller text for the link message */ } """ # --- Build the Gradio UI with Blocks --- with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app: gr.HTML("