Update app.py

app.py

@@ -471,6 +471,11 @@ footer {
 .btn {
     display: flex;
 }
+
+/* Added background gradient for a more colorful look */
+.gradio-container {
+    background: linear-gradient(to right, #ffecd2, #fcb69f) !important;
+}
 """])
 
 
@@ -492,7 +497,6 @@ def easy_scaler(img):
     return upscaler(img)
 
 def handle_upscaler(img):
-    
     w, h = img.size
     if w*h > 2 * (10 ** 6):
         return hard_scaler(img)
@@ -538,7 +542,6 @@ def upscaler(
     )
 
     log(f'RET upscaler')
-
     return enhanced_image
 
 def get_tensor_length(tensor):
@@ -551,7 +554,7 @@ def get_tensor_length(tensor):
 def _summarize(text):
     log(f'CALL _summarize')
     prefix = "summarize: "
-    toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
+    toks = tokenizer.encode(prefix + text, return_tensors="pt", truncation=False)
     gen = model.generate(
         toks,
         length_penalty=0.1,
@@ -572,9 +575,7 @@ def summarize(text, max_words=100):
     if words_length >= 510:
         while words_length >= 510:
             words = text.split()
-            sum = _summarize(
-                " ".join(words[0:510])
-            ) + " ".join(words[510:])
+            summ = _summarize(" ".join(words[0:510])) + " ".join(words[510:])
             if summ == text:
                 return text
             text = summ
@@ -597,7 +598,6 @@ def generate_random_string(length):
 def add_text_above_image(img,top_title=None,bottom_title=None):
 
     w, h = img.size
-    
     draw = ImageDraw.Draw(img,mode="RGBA")
 
     labels_distance = 1/3
@@ -610,7 +610,15 @@ def add_text_above_image(img,top_title=None,bottom_title=None):
         x = math.ceil((w - textwidth) / 2)
         y = h - (textheight * rows / 2) - (h / 2)
         y = math.ceil(y - (h / 2 * labels_distance))
-        draw.text((x, y), top_title, (255,255,255), font=font, spacing=2, stroke_width=math.ceil(textheight/20), stroke_fill=(0,0,0))
+        draw.text(
+            (x, y),
+            top_title,
+            (255,255,255),
+            font=font,
+            spacing=2,
+            stroke_width=math.ceil(textheight/20),
+            stroke_fill=(0,0,0)
+        )
 
     if bottom_title:
         rows = len(bottom_title.split("\n"))
@@ -620,7 +628,15 @@ def add_text_above_image(img,top_title=None,bottom_title=None):
         x = math.ceil((w - textwidth) / 2)
         y = h - (textheight * rows / 2) - (h / 2)
         y = math.ceil(y + (h / 2 * labels_distance))
-        draw.text((x, y), bottom_title, (0,0,0), font=font, spacing=2, stroke_width=math.ceil(textheight/20), stroke_fill=(255,255,255))
+        draw.text(
+            (x, y),
+            bottom_title,
+            (0,0,0),
+            font=font,
+            spacing=2,
+            stroke_width=math.ceil(textheight/20),
+            stroke_fill=(255,255,255)
+        )
     
     return img
 
@@ -769,10 +785,6 @@ class BaseError(Exception):
     """
 
     def __init__(self, val, message):
-        """
-        @param val: actual value
-        @param message: message shown to the user
-        """
         self.val = val
         self.message = message
         super().__init__()
@@ -866,7 +878,6 @@ class RequestError(Exception):
         return self.message
 
 
-
 class TooManyRequests(Exception):
     """
     exception thrown if an error occurred during the request call, e.g a connection problem.
@@ -884,6 +895,7 @@ class TooManyRequests(Exception):
     def __str__(self):
         return self.message
 
+
 class ServerException(Exception):
     """
     Default YandexTranslate exception from the official website
@@ -913,14 +925,7 @@ def is_empty(text: str) -> bool:
 
 def request_failed(status_code: int) -> bool:
     """Check if a request has failed or not.
-    A request is considered successfull if the status code is in the 2** range.
-
-    Args:
-        status_code (int): status code of the request
-
-    Returns:
-        bool: indicates request failure
-    """
+    A request is considered successful if the status code is in the 2** range."""
     if status_code > 299 or status_code < 200:
         return True
     return False
@@ -936,12 +941,10 @@ def is_input_valid(
     @param text: text to translate
     @return: bool
     """
-
     if not isinstance(text, str):
         raise NotValidPayload(text)
     if max_chars and (not min_chars <= len(text) < max_chars):
         raise NotValidLength(text, min_chars, max_chars)
-
     return True
 
 class BaseTranslator(ABC):
@@ -1037,7 +1040,7 @@ class BaseTranslator(ABC):
         """
         check if the language is supported by the translator
         @param language: a string for 1 language
-        @return: bool or raise an Exception
+        @return: bool
         """
         if (
             language == "auto"
@@ -1061,12 +1064,10 @@ class BaseTranslator(ABC):
 
     def _read_docx(self, f: str):
         import docx2txt
-
         return docx2txt.process(f)
 
     def _read_pdf(self, f: str):
         import pypdf
-
         reader = pypdf.PdfReader(f)
         page = reader.pages[0]
         return page.extract_text()
@@ -1090,7 +1091,6 @@ class BaseTranslator(ABC):
 
         if ext == ".docx":
             text = self._read_docx(f=str(path))
-
         elif ext == ".pdf":
             text = self._read_pdf(f=str(path))
         else:
@@ -1176,6 +1176,7 @@ class GoogleTranslator(BaseTranslator):
                 element = soup.find(self._element_tag, self._alt_element_query)
                 if not element:
                     raise TranslationNotFound(text)
+
             if element.get_text(strip=True) == text.strip():
                 to_translate_alpha = "".join(
                     ch for ch in text.strip() if ch.isalnum()
@@ -1246,22 +1247,20 @@ def translate(txt,to_lang="en",from_lang="auto"):
     return translation.lower()
 
 def handle_generation(h,w,d):
-
     log(f'CALL handle_generate')
-
     difficulty_points = 0
 
-    toks_len = get_tensor_length(tokenizer.encode( d, return_tensors="pt", truncation=False))
+    toks_len = get_tensor_length(tokenizer.encode(d, return_tensors="pt", truncation=False))
     if toks_len > 500:
-        difficulty_points + 2
+        difficulty_points += 2
     elif toks_len > 50:
-        difficulty_points + 1
+        difficulty_points += 1
 
     pxs = h*w
     if pxs > 2 * (10 ** 6):
-        difficulty_points + 2
+        difficulty_points += 2
     elif pxs > 1 * (10 ** 6):
-        difficulty_points + 1
+        difficulty_points += 1
 
     if difficulty_points < 2:
         return easy_generation(h,w,d)
@@ -1283,7 +1282,6 @@ def hard_generation(h,w,d):
     return generation(h,w,d)
 
 def generation(h,w,d):
-    
     if len(d) > 0:
         d = re.sub(r",( ){1,}",". ",d)
         d_lines = re.split(r"([\n]){1,}", d)
@@ -1295,12 +1293,11 @@ def generation(h,w,d):
         d = " ".join(d_lines)
     
         d = re.sub(r"([ \t]){1,}", " ", d).lower().strip()
-        d = d if d == "" else summarize(translate(d),max_words=50)
+        d = d if d == "" else summarize(translate(d), max_words=50)
         d = re.sub(r"([ \t]){1,}", " ", d)
         d = re.sub(r"(\. \.)", ".", d)
 
     neg = f"Textual, Text, Signs, Labels, Titles, Unreal, Exceptional, Irregular, Unusual, Blurry, Smoothed, Polished, Worst Quality, Worse Quality, Painted, Movies Quality."
-    q = "\""
     pos = f'Accurate, Detailed, Realistic.{ "" if d == "" else " " + d }'
 
     print(f"""
@@ -1321,15 +1318,15 @@ def generation(h,w,d):
         max_sequence_length=seq,
         generator=torch.Generator(device).manual_seed(random.randint(0, MAX_SEED))
     ).images[0]
-    
     return img
-   
+
 # entry
 
 if __name__ == "__main__":
-    with gr.Blocks(theme=gr.themes.Citrus(),css=css) as demo:
+    # Changed the theme to a more colorful one and updated the title
+    with gr.Blocks(theme=gr.themes.Soft(primary_hue="lime"), css=css) as demo:
         gr.Markdown(f"""
-            # Text-to-Image generator
+            # Multilingual Images
         """)
         gr.Markdown(f"""
             ### Realistic. Upscalable. Multilingual.
@@ -1337,7 +1334,6 @@ if __name__ == "__main__":
         
         with gr.Row():
             with gr.Column(scale=2):
-                
                 height = gr.Slider(
                     label="Height (px)",
                     minimum=512,
@@ -1345,7 +1341,6 @@ if __name__ == "__main__":
                     step=16,
                     value=1024,
                 )
-
                 width = gr.Slider(
                     label="Width (px)",
                     minimum=512,
@@ -1354,7 +1349,7 @@ if __name__ == "__main__":
                     value=1024,
                 )
 
-                run = gr.Button("Generate",elem_classes="btn")
+                run = gr.Button("Generate", elem_classes="btn")
 
                 top = gr.Textbox(
                     placeholder="Top title",
@@ -1370,21 +1365,51 @@ if __name__ == "__main__":
                 )
 
                 data = gr.Textbox(
-                    placeholder="Input data",
+                    placeholder="Input data (text/prompt)",
                     value="",
                     container=False,
                     max_lines=100
                 )
 
             with gr.Column():
-                cover = gr.Image(interactive=False,container=False,elem_classes="image-container", label="Result", show_label=True, type='pil', show_share_button=False)
-                upscale_now = gr.Button("Upscale x2",elem_classes="btn")
-                add_titles = gr.Button("Add title(s)",elem_classes="btn")
+                cover = gr.Image(
+                    interactive=False,
+                    container=False,
+                    elem_classes="image-container",
+                    label="Result",
+                    show_label=True,
+                    type='pil',
+                    show_share_button=False
+                )
+                upscale_now = gr.Button("Upscale x2", elem_classes="btn")
+                add_titles = gr.Button("Add title(s)", elem_classes="btn")
+
+        gr.Markdown("---")
+
+        # Bottom row explanation or details
+        gr.Markdown("""
+        ## 기능 세부 설명 (Features)
+        1. **텍스트 입력**: 다양한 언어로 텍스트 입력 시 자동 번역 및 요약 후 이미지를 생성합니다.  
+        2. **이미지 크기 조절**: 슬라이더를 통해 생성될 이미지의 크기를 지정할 수 있습니다.  
+        3. **오버레이 텍스트**: 이미지 위에 상단/하단 타이틀을 추가하여 간단한 라벨링이 가능합니다.  
+        4. **고화질 업스케일**: 'Upscale x2' 버튼을 통해 이미지 품질을 향상시킬 수 있습니다.  
+        5. **자동 생성 난이도 조절**: 입력 텍스트와 이미지 크기에 따라 GPU 사용시간이 자동으로 조절됩니다.  
+        ---  
+        """)
+
+        gr.Markdown("""
+        ### 이용 가이드
+        1. 원하는 이미지 사이즈와 텍스트(다국어 가능)를 입력 후 **Generate** 버튼을 클릭하세요.  
+        2. 결과 이미지를 확인한 뒤, **Upscale x2** 기능을 사용하면 더 높은 해상도로 업스케일 됩니다.  
+        3. **Add title(s)** 버튼을 사용하면 이미지 위에 타이틀을 쉽게 추가할 수 있습니다.  
+        4. 기능 세부 설명은 상단의 입력항목들을 통해 테스트해 보시면 됩니다.
+        """)
 
+        # Event wiring
         gr.on(
             triggers=[run.click],
             fn=handle_generation,
-            inputs=[height,width,data],
+            inputs=[height, width, data],
             outputs=[cover]
         )
         upscale_now.click(
@@ -1394,7 +1419,7 @@ if __name__ == "__main__":
         )
         add_titles.click(
             fn=add_text_above_image,
-            inputs=[cover,top,bottom],
+            inputs=[cover, top, bottom],
             outputs=[cover]
         )