Update app.py

app.py

@@ -39,7 +39,7 @@ def translate(text,lang):
         'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15',
         'Mozilla/5.0 (Macintosh; Intel Mac OS X 13_1) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15'
     ]
-    url = f'https://www.google.com/search?q=translate to {lang}: {text}'
+    url = f'https://www.google.com/search?q={lang}: {text}'
     print(url)
     resp = requests.get(
         url = url,
@@ -69,47 +69,23 @@ def generate_random_string(length):
 def Piper(_do):
     return pipe(
         _do,
-        height=480,
-        width=480,
+        height=512,
+        width=512,
         negative_prompt="",
         num_inference_steps=100,
-        guidance_scale=9
+        guidance_scale=10
     )
 
-def infer(prompt1,prompt2,prompt3,prompt4):
+def infer(prompt):
     name = generate_random_string(12)+".png"
-
-    if prompt4 == None:
-        prompt4 = " without text nor logo"
-    else:
-        prompt4 = " without text nor logo " + " nor ".join([translate(v,"english") for v in prompt4])
-
-    prompt4 = prompt4.upper()
-    
-    if prompt1 == None:
-        prompt1 = "focused elements" + prompt4
-    else:
-        prompt1 = ", ".join([ "primary "+(translate(v,"english")+prompt4) for v in prompt1 ])
-    
-    if prompt2 == None:
-        prompt2 =  "secondary elements" + prompt4
-    else:
-        prompt2 = ", ".join([ "secondary "+(translate(v,"english")+prompt4) for v in prompt2 ])
-    
-    if prompt3 == None:
-        prompt3 = "event" + prompt4
-    else:
-        prompt3 = ", ".join([ (translate(v,"english")+" "+prompt4) for v in prompt3 ])
-    
-    _do = f'Rational fusion, from, {prompt3}, {prompt2}, {prompt1}'.upper()
-    
+    _do = f'true {prompt1}:'.upper()
     image = Piper(_do).images[0].save(name)
     return name
 
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 13cm;
+    max-width: 15cm;
 }
 #image-container {
     aspect-ratio: 1 / 1;
@@ -124,55 +100,33 @@ if torch.cuda.is_available():
 else:
     power_device = "CPU"
 
+def limiter(txt):
+    return (re.sub(f'[{string.punctuation}]', '', re.sub('[\s+]', ' ', txt)).lower().strip())[:38]
+
 with gr.Blocks(theme=gr.themes.Soft(),css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
             # Image Generator
             Currently running on {power_device}.
         """)
-        with gr.Column():
-            with gr.Row():
-                prompt1 = gr.Dropdown(
-                    multiselect=True,
-                    allow_custom_value=True,
-                    max_choices=1,
-                    label="Foreground Elements",
-                    show_label=True,
-                    container=True
-                )
-            with gr.Row():
-                prompt2 = gr.Dropdown(
-                    multiselect=True,
-                    allow_custom_value=True,
-                    max_choices=2,
-                    label="Background Elements",
-                    show_label=True,
-                    container=True
-                )
-            with gr.Row():
-                prompt3 = gr.Dropdown(
-                    multiselect=True,
-                    allow_custom_value=True,
-                    max_choices=1,
-                    label="Background Events",
-                    show_label=True,
-                    container=True
-                )
-            with gr.Row():
-                prompt4 = gr.Dropdown(
-                    multiselect=True,
-                    allow_custom_value=True,
-                    max_choices=2,
-                    label="Forbidden Elements/Events",
-                    show_label=True,
-                    container=True
-                )
-            with gr.Row():
-                run_button = gr.Button("Run")
+        with gr.Row():
+            prompt = gr.Textbox(
+                placeholder="Describe the photo...",
+                container=False,
+                rtl=True,
+                max_lines=1
+            )
+        with gr.Row():
+            run_button = gr.Button("Run")
         result = gr.Image(elem_id="image-container", label="Result", show_label=False, type='filepath')
+    prompt.change(
+        fn = limiter,
+        inputs = [prompt],
+        outputs = [prompt],
+    )
     run_button.click(
         fn = infer,
-        inputs = [prompt1,prompt2,prompt3,prompt4],
+        inputs = [prompt],
         outputs = [result]
     )