Update app.py

app.py

@@ -56,26 +56,29 @@ def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="
         return None
 
     if lora_id.strip() == "" or lora_id == None:
-        lora_id = "black-forest-labs/FLUX.1-dev" 
+        lora_id = "black-forest-labs/FLUX.1-dev"
 
     key = random.randint(0, 999)
 
     API_URL = "https://api-inference.huggingface.co/models/"+ lora_id.strip()
-    
+
     API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN")])
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
     
     # prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     # print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
+    prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
+    print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
 
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
-    # print(f'\033[1mGeneration {key}:\033[0m {prompt}')
+    print(f'\033[1mGeneration {key}:\033[0m {prompt}')
 
     # If seed is -1, generate a random seed and use it
     if seed == -1:
         seed = random.randint(1, 1000000000)
-            
-        payload = {
+
+    # Prepare the payload for the API call, including width and height
+    payload = {
         "inputs": prompt,
         "is_negative": is_negative,
         "steps": steps,
@@ -95,23 +98,23 @@ def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="
         if response.status_code == 503:
             raise gr.Error(f"{response.status_code} : The model is being loaded")
         raise gr.Error(f"{response.status_code}")
-    
+
     try:
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
         print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
-        return image, seed, seed
+        return image, seed
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
         return None
 
-
 examples = [
     "a beautiful woman with blonde hair and blue eyes",
     "a beautiful woman with brown hair and grey eyes",
     "a beautiful woman with black hair and brown eyes",
 ]
 
+
 css = """
 #app-container {
     max-width: 896px;
@@ -176,14 +179,13 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                 
             with gr.Row():
                 image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
-        
-        gr.Examples(
-            examples = examples,
-            inputs = [text_prompt],
-        )
-
-        
-        text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, seed_output])
+    
+            gr.Examples(
+                examples = examples,
+                inputs = [text_prompt],
+            )
+            
+    text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, seed_output])
 
 
 app.launch(show_api=False, share=False)