Update app.py

app.py

@@ -37,6 +37,7 @@ def enable_lora(lora_add):
 
 async def generate_image(
     prompt:str,
+    negative_prompt:str,
     model:str,
     lora_word:str,
     width:int=768,
@@ -50,12 +51,14 @@ async def generate_image(
     seed = int(seed)
     print(f'prompt:{prompt}')
     
-    text = str(translator.translate(prompt, 'English')) + lora_word
+    text = str(translator.translate(prompt, 'English')) + "," + lora_word
+    negative_prompt = str(translator.translate(negative_prompt, 'English'))
 
     client = AsyncInferenceClient()
 
     image = await client.text_to_image(
         prompt=text,
+        negative_prompt=negative_prompt,
         height=height,
         width=width,
         guidance_scale=scales,
@@ -70,6 +73,7 @@ async def gen(
     prompt:str,
     lora_add:str="",
     lora_word:str="",
+    negative_prompt:str="",
     width:int=768,
     height:int=1024,
     scales:float=3.5,
@@ -79,22 +83,22 @@ async def gen(
 ):
     model = enable_lora(lora_add)
     print(model)
-    image, seed = await generate_image(prompt,model,lora_word,width,height,scales,steps,seed)
+    image, seed = await generate_image(prompt,negative_prompt,model,lora_word,width,height,scales,steps,seed)
     return image, seed
      
 examples = [
     ["A cartoon-style blonde European-American woman wearing sunglasses stood in front of the triumphant door to take a selfie, the upper bodyartistic style","Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration","artistic style blends elements of reality and illustration"],
     ["A cartoon style European woman wearing glasses is eating a table of seafood,including lobster,oysters,and other shellfish,in a well lit modern restaurant. The background of the restaurant is very blurry,and she is holding the utensils ready to eat. There is a glass of red wine and various dishes on the table. The illustrations contrast with the real food and environment,creating a unique mixed media effect and high angle perspective","Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration","artistic style blends elements of reality and illustration"],
-    ["A cartoon style European man opens his hands and takes a selfie under the Sydney Opera House","Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration","artistic style blends elements of reality and illustration"],
-    ["Against the backdrop of the Eiffel Tower, a cartoon style European woman wearing a delicate white floral dress stands there, with the iconic building of the tower clearly visible under the azure sky, capturing the romantic charm of Paris. When she takes photos against this stunning background, her flowing hair adds a dreamy atmosphere","Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration","artistic style blends elements of reality and illustration"]
+    ["photograph, background of Earth from space, red car on the Moon watching Earth","martintomov/retrofuturism-flux","retrofuturism"],
+    ["red sports car with `SPACE RIDER` on its frontal hood, frontal view with a person visible inside through the windshield, car is emerging from a circular radiant portal with bright orange and red hues against a dark background, sense of high-speed motion and a tunnel-like effect","martintomov/retrofuturism-flux","retrofuturism"]
 ]
 
 
 # Gradio Interface
 
 with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
-    gr.HTML("<h1><center>Flux Labs</center></h1>")
-    gr.HTML("<p><center>Add the LoRA model on the menu</center></p>")
+    gr.HTML("<h1><center>Flux Lab Light</center></h1>")
+    gr.HTML("<p><center>Powered By HF Inference API</center></p>")
     with gr.Row():
         with gr.Column(scale=4):
             with gr.Row():
@@ -139,6 +143,11 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
                     step=1,
                     value=-1,
                 )
+                negative_prompt = gr.Textbox(
+                    label="Negative prompt",
+                    lines=1,
+                    value="low quality"
+                )
                 lora_add = gr.Textbox(
                     label="Add Flux LoRA",
                     info="Copy the HF LoRA model name here",
@@ -171,6 +180,7 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
             prompt,
             lora_add,
             lora_word,
+            negative_prompt,
             width, 
             height, 
             scales,