Update app.py

app.py

@@ -1431,31 +1431,26 @@ def translate(txt,to_lang="en",from_lang="auto"):
     return translation.lower()
 
 @spaces.GPU(duration=300)
-def handle_generation(artist,song,lyrics,h,w):
+def handle_generation(h,w,d):
 
     log(f'CALL handle_generate')
-    
-    pos_artist = re.sub(r"([ \t]){1,}", " ", artist).upper().strip()
-    
-    pos_song = re.sub(r"([ \t]){1,}", " ", song).lower().strip()
-    pos_song = ' '.join(word[0].upper() + word[1:] for word in pos_song.split())
 
-    lyrics = re.sub(r",( ){1,}",". ",lyrics)
-    lyrics_lines = re.split(r"([\n]){1,}", lyrics)
+    d = re.sub(r",( ){1,}",". ",d)
+    d_lines = re.split(r"([\n]){1,}", d)
     
-    for line_index in range(len(lyrics_lines)):
-        lyrics_lines[line_index] = lyrics_lines[line_index].strip()
-        if re.sub(r'[\.]$', '', lyrics_lines[line_index]) == lyrics_lines[line_index]:
-            lyrics_lines[line_index] = lyrics_lines[line_index].strip() + "."
-    lyrics = " ".join(lyrics_lines)
+    for line_index in range(len(d_lines)):
+        d_lines[line_index] = d_lines[line_index].strip()
+        if re.sub(r'[\.]$', '', d_lines[line_index]) == d_lines[line_index]:
+            d_lines[line_index] = d_lines[line_index].strip() + "."
+    d = " ".join(d_lines)
 
-    pos_lyrics = re.sub(r"([ \t]){1,}", " ", lyrics).lower().strip()
-    pos_lyrics = pos_lyrics if pos_lyrics == "" else summarize(translate(pos_lyrics))
-    pos_lyrics = re.sub(r"([ \t]){1,}", " ", pos_lyrics).lower().strip()
+    pos_d = re.sub(r"([ \t]){1,}", " ", d).lower().strip()
+    pos_d = pos_d if pos_d == "" else summarize(translate(pos_d))
+    pos_d = re.sub(r"([ \t]){1,}", " ", pos_d).lower().strip()
 
     neg = f"Textual, Text, Distorted, Fake, Discontinuous, Blurry, Doll-Like, Overly Plastic, Low Quality, Paint, Smoothed, Artificial, Phony, Gaudy, Digital Effects."
     q = "\""
-    pos = f'HQ Hyper-realistic professional photograph{ pos_lyrics if pos_lyrics == "" else ": " + q + pos_lyrics + q }.'
+    pos = f'HQ Hyper-realistic professional photograph{ pos_d if pos_d == "" else ": " + pos_d }.'
 
     print(f"""
         Positive: {pos}
@@ -1463,57 +1458,45 @@ def handle_generation(artist,song,lyrics,h,w):
         Negative: {neg}
     """)
 
-    return add_song_cover_text(pipe_generate_image(pos,neg,h,w),pos_artist,pos_song,h,w)
+    return pipe_generate_image(pos,neg,h,w)
     
 # entry
 
 if __name__ == "__main__":
     with gr.Blocks(theme=gr.themes.Citrus(),css=css) as demo:
         gr.Markdown(f"""
-            # Text to Image
+            # Text-to-Image generator
         """)
         gr.Markdown(f"""
-            ## Multilingual. Realistic. Top & Bottom titles.
+            ### Realistic. Upscalable. Multilingual.
         """)
         
         with gr.Row():
-            with gr.Column(scale=2):
-                artist = gr.Textbox(
-                        placeholder="Top title",
-                        value="",
-                        container=False,
-                        max_lines=1
-                )
-                song = gr.Textbox(
-                        placeholder="Bottom title",
-                        value="",
-                        container=False,
-                        max_lines=1
-                )
+            with gr.Column(scale=3):
                 
-                run = gr.Button("Generate",elem_classes="btn")
-                
-                width = gr.Slider(
-                    label="Width (px)",
+                height = gr.Slider(
+                    label="Height (px)",
                     minimum=512,
                     maximum=2048,
                     step=16,
                     value=1024,
                 )
 
-                height = gr.Slider(
-                    label="Height (px)",
+                width = gr.Slider(
+                    label="Width (px)",
                     minimum=512,
                     maximum=2048,
                     step=16,
                     value=1024,
                 )
+
+                run = gr.Button("Generate",elem_classes="btn")
     
-                lyrics = gr.Textbox(
-                    placeholder="Textual data (multilingual)",
+                data = gr.Textbox(
+                    placeholder="Input data",
                     value="",
                     container=False,
-                    max_lines=1000
+                    max_lines=100
                 )
 
             with gr.Column():
@@ -1522,7 +1505,7 @@ if __name__ == "__main__":
 
         run.click(
             fn=handle_generation,
-            inputs=[artist,song,lyrics,height,width],
+            inputs=[height,width,data],
             outputs=[cover]
         )
         upscale_now.click(