Update app.py

app.py

@@ -85,29 +85,32 @@ def translate_file(file, src_lang, tgt_lang, max_length, temperature):
     except Exception as e:
         return f"File translation error: {e}"
 
-# App UI
-with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+# Custom CSS to improve UI
+gradio_style = """
+.gr-button { border-radius: 12px !important; padding: 10px 20px !important; font-weight: bold; }
+textarea, input[type=text] { border: 2px solid #00ADB5 !important; border-radius: 10px; transition: 0.2s; }
+textarea:focus, input[type=text]:focus { border-color: #FF5722 !important; box-shadow: 0 0 8px #FF5722 !important; }
+"""
+
+with gr.Blocks(css=gradio_style, theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    <div style="text-align: center; font-size: 28px; font-weight: bold; margin-bottom: 20px;">
-    🌍 PolyLinguaAI: Translate Across Worlds
-    </div>
-    <div style="text-align: center; font-size: 16px; margin-bottom: 40px;">
-    Translate between 12+ global languages using the powerful NLLB model.
-    </div>
+    # 🌍 PolyLinguaAI: Translate Across Worlds
+    Translate instantly between 12+ languages using Facebook's NLLB model.
     """)
 
     with gr.Tab("🌐 Text Translator"):
         with gr.Row():
-            src_lang = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="From", value="English")
-            swap = gr.Button("⇄", size="sm")
-            tgt_lang = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="To", value="Korean")
+            src_lang = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="🌐 From", value="English")
+            swap = gr.Button("⇄")
+            tgt_lang = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="🎯 To", value="Korean")
 
-        input_text = gr.Textbox(lines=5, label="Input Text", placeholder="Type or paste text to translate...")
-        output_text = gr.Textbox(lines=5, label="Translated Output", interactive=False)
+        with gr.Row():
+            input_text = gr.Textbox(lines=3, label="✍️ Input Text")
+            output_text = gr.Textbox(lines=3, label="📤 Translated Output", interactive=False)
 
         with gr.Row():
-            translate = gr.Button("🔁 Translate", variant="primary")
-            clear = gr.Button("🧼 Clear")
+            translate = gr.Button("🚀 Translate", variant="primary")
+            clear = gr.Button("🧽 Clear")
 
         with gr.Accordion("⚙️ Advanced Settings", open=False):
             max_length = gr.Slider(10, 512, value=128, step=1, label="Max Length")
@@ -117,33 +120,34 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
             history_json = gr.JSON(label="Recent Translations")
             with gr.Row():
                 refresh = gr.Button("🔄 Refresh")
-                clear_history = gr.Button("🧹 Clear")
+                clear_history = gr.Button("🧹 Clear History")
 
     with gr.Tab("📁 File Translator"):
-        file_input = gr.File(label="Upload .txt File", file_types=[".txt"])
-        file_src = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="From", value="English")
-        file_tgt = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="To", value="Korean")
-        file_result = gr.Textbox(label="Translated File Output")
-        file_translate = gr.Button("Translate File", variant="primary")
+        file_input = gr.File(label="📂 Upload .txt File", file_types=[".txt"])
+        file_src = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="📌 From", value="English")
+        file_tgt = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="📌 To", value="Korean")
+        file_translate = gr.Button("📄 Translate File", variant="primary")
+        file_result = gr.Textbox(label="📑 File Output", lines=10, interactive=False)
 
-        with gr.Accordion("⚙️ File Settings", open=False):
+        with gr.Accordion("⚙️ Advanced Settings", open=False):
             f_max_length = gr.Slider(10, 512, value=128, step=1, label="Max Length")
             f_temp = gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature")
 
+    # Events
     translate.click(cached_translate, [input_text, src_lang, tgt_lang, max_length, temperature], output_text)
     clear.click(lambda: ("", ""), None, [input_text, output_text])
     swap.click(swap_langs, [src_lang, tgt_lang], [src_lang, tgt_lang])
     refresh.click(lambda: history.get(), None, history_json)
     clear_history.click(lambda: history.clear() or [], None, history_json)
     file_translate.click(lambda file, src, tgt, ml, t: translate_file(file.read(), src, tgt, ml, t),
-                         [file_input, file_src, file_tgt, f_max_length, f_temp],
-                         file_result)
+                         [file_input, file_src, file_tgt, f_max_length, f_temp], file_result)
 
     gr.Markdown(f"""
-    <hr/>
-    <div style="font-size: 14px; text-align: center;">
-    ⚙️ Model: <code>{model_name}</code> | Device: <code>{device}</code> | Cache: 512 entries
-    </div>
+    ### 🛠 Model Info
+    - Model: `{model_name}`
+    - Device: `{device}`
+    - Cached Translations: 512
     """)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch(share=True)