Update app.py

app.py

@@ -6,14 +6,14 @@ from functools import lru_cache
 import torch
 import requests
 
-# Language Codes
+# Language codes
 LANGUAGE_CODES = {
     "English": "eng_Latn", "Korean": "kor_Hang", "Japanese": "jpn_Jpan", "Chinese": "zho_Hans",
     "Spanish": "spa_Latn", "French": "fra_Latn", "German": "deu_Latn", "Russian": "rus_Cyrl",
     "Portuguese": "por_Latn", "Italian": "ita_Latn", "Burmese": "mya_Mymr", "Thai": "tha_Thai"
 }
 
-# Translation History
+# Translation History class
 class TranslationHistory:
     def __init__(self):
         self.history = []
@@ -25,21 +25,19 @@ class TranslationHistory:
         })
         if len(self.history) > 100:
             self.history.pop()
-    def get(self):
-        return self.history
-    def clear(self):
-        self.history = []
+    def get(self): return self.history
+    def clear(self): self.history = []
 
 history = TranslationHistory()
 
-# Load Translation Model
+# Load translation model
 model_name = "facebook/nllb-200-distilled-600M"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
-# Translation Function
+# Translate function
 @lru_cache(maxsize=512)
 def cached_translate(text, src_lang, tgt_lang, max_length=128, temperature=0.7):
     if not text.strip(): return ""
@@ -48,7 +46,8 @@ def cached_translate(text, src_lang, tgt_lang, max_length=128, temperature=0.7):
     input_tokens = tokenizer(text, return_tensors="pt", padding=True)
     input_tokens = {k: v.to(device) for k, v in input_tokens.items()}
     forced_bos_token_id = tokenizer.convert_tokens_to_ids(tgt_code)
-    output = model.generate(**input_tokens,
+    output = model.generate(
+        **input_tokens,
         forced_bos_token_id=forced_bos_token_id,
         max_length=max_length, temperature=temperature,
         num_beams=5, early_stopping=True
@@ -65,9 +64,9 @@ def translate_file(file, src_lang, tgt_lang, max_length, temperature):
     except Exception as e:
         return f"File translation error: {e}"
 
-# Summarization
+# Hugging Face Summarization API
 API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
-HF_API_KEY = os.environ.get("HF_API_KEY", "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")
+HF_API_KEY = os.environ.get("HF_API_KEY")
 headers = {"Authorization": f"Bearer {HF_API_KEY}"}
 
 def summarize_text(text, max_length):
@@ -80,13 +79,14 @@ def summarize_text(text, max_length):
     result = response.json()
     return result[0]["summary_text"] if isinstance(result, list) else "Error: " + str(result)
 
-# UI Styling
+# UI Style
 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; }
 """
 
+# Gradio App
 with gr.Blocks(css=gradio_style, theme=gr.themes.Soft()) as demo:
     gr.Markdown("## 🤖 AI Toolbox: Translate & Summarize")
 
@@ -95,30 +95,35 @@ with gr.Blocks(css=gradio_style, theme=gr.themes.Soft()) as demo:
             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")
+
         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")
+
         with gr.Accordion("⚙️ Advanced Settings", open=False):
             max_length = gr.Slider(10, 512, value=128, step=1, label="Max Length")
             temperature = gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature")
+
         with gr.Accordion("📜 Translation History", open=False):
             history_json = gr.JSON(label="Recent Translations")
             with gr.Row():
                 refresh = gr.Button("🔄 Refresh")
                 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_translate = gr.Button("📄 Translate File", variant="primary")
-            file_result = gr.Textbox(label="📑 File Output", lines=10, interactive=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")
+    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_translate = gr.Button("📄 Translate File", variant="primary")
+        file_result = gr.Textbox(label="📑 File Output", lines=10, interactive=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")
 
     with gr.Tab("📝 Text Summarizer"):
         summary_input = gr.Textbox(lines=5, label="📚 Enter text to summarize")
@@ -126,7 +131,7 @@ with gr.Blocks(css=gradio_style, theme=gr.themes.Soft()) as demo:
         summary_output = gr.Textbox(label="🧾 Summary", lines=5, interactive=False)
         summary_btn = gr.Button("🧠 Summarize")
 
-    # Button events
+    # Event Hooks
     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(lambda s, t: (t, s), [src_lang, tgt_lang], [src_lang, tgt_lang])
@@ -138,9 +143,9 @@ with gr.Blocks(css=gradio_style, theme=gr.themes.Soft()) as demo:
 
     gr.Markdown(f"""
     ### 🔍 Info
-    - Translator Model: `{model_name}` on `{device}`
-    - Summarizer Model: `facebook/bart-large-cnn`
-    - HF API Key: {'Loaded ✅' if HF_API_KEY else 'Missing ❌'}
+    - Translator: `{model_name}` on `{device}`
+    - Summarizer: `facebook/bart-large-cnn`
+    - API Token Status: {'✅ Loaded' if HF_API_KEY else '❌ Not Found'}
     """)
 
 if __name__ == "__main__":