Update app.py

app.py

@@ -5,16 +5,29 @@ import base64
 import threading
 from huggingface_hub import login, hf_hub_download
 
+# Setup HuggingFace authentication and download
 login(token=os.environ["HF_TOKEN"])
 repo_id = os.environ["REPO_ID"]
 
+# Torch optimizations
 import torch
 torch.set_num_threads(1)
 if torch.cuda.is_available():
     torch.backends.cudnn.benchmark = True
 
+# Download generation logic and dependencies from private repo
 try:
-    files_to_download = ["generate.py", "model.py", "hfmod_sft.py"]
+    # Download all required files
+    files_to_download = [
+        "generate.py", 
+        "model.py", 
+        "hfmod_sft.py",
+        "tokenizer.json",
+        "tokenizer_config.json", 
+        "special_tokens_map.json",
+        "vocab.json",
+        "merges.txt"
+    ]
     
     for filename in files_to_download:
         try:
@@ -31,9 +44,11 @@ try:
 except Exception as e:
     print(f"Error downloading files: {e}")
 
+# Import the generation function
 sys.path.append('.')
 from generate import chat_interface, init_model
 
+# Logo and favicon setup
 LOGO_PATH = "static/logo.png"
 if os.path.isfile(LOGO_PATH):
     with open(LOGO_PATH, "rb") as f:
@@ -45,6 +60,7 @@ else:
     FAVICON_HTML = ''
 
 def init_translation_model():
+    """Initialize the translation model"""
     success = init_model()
     if success:
         print("Model loaded successfully!")
@@ -52,10 +68,19 @@ def init_translation_model():
         print("Failed to load model")
 
 def translate_text(source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens):
-    return chat_interface(source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens)
+    """Handle translation requests"""
+    return chat_interface(
+        source_text, source_lang, target_lang, 
+        temperature, top_k, repetition_penalty, max_tokens
+    )
 
+# Language options
 languages = ["English", "Hindi", "Bengali", "Tamil", "Telugu", "Kannada", "Panjabi"]
 
+# Custom CSS
+css_path = "static/style.css"
+custom_css = open(css_path, encoding="utf-8").read() if os.path.isfile(css_path) else ""
+
 theme_lock_css = """
 .gradio-container .theme-toggle,
 .gradio-container button[aria-label*="theme"],
@@ -73,9 +98,14 @@ body, .gradio-container {
 }
 """
 
-combined_css = theme_lock_css
+combined_css = custom_css + theme_lock_css
 
-locked_theme = gr.themes.Monochrome(primary_hue="blue", secondary_hue="slate", neutral_hue="slate").set(
+# Theme configuration
+locked_theme = gr.themes.Monochrome(
+    primary_hue="blue",
+    secondary_hue="slate",
+    neutral_hue="slate"
+).set(
     background_fill_primary="#0a1628",
     background_fill_secondary="#1f2937", 
     block_background_fill="#374151",
@@ -133,6 +163,7 @@ with gr.Blocks(
             """)
 
     with gr.Row(equal_height=False):
+        # Settings panel
         with gr.Column(scale=1.5, elem_classes="settings-panel"):
             gr.Markdown("## Translation Settings")
             with gr.Row():
@@ -148,11 +179,22 @@ with gr.Blocks(
 
         with gr.Column(scale=2, elem_classes="translation-card"):
             gr.Markdown("## Translation Interface")
-            source_text = gr.Textbox(label="Enter text to translate", placeholder="Type or paste your text here", lines=6, max_lines=12)
+            source_text = gr.Textbox(
+                label="Enter text to translate", 
+                placeholder="Type or paste your text here", 
+                lines=6, 
+                max_lines=12
+            )
             with gr.Row():
                 translate_btn = gr.Button("Translate", variant="primary", size="lg")
                 clear_btn = gr.Button("Clear All", variant="secondary", size="lg")
-            translated_text = gr.Textbox(label="Translation Result", lines=6, max_lines=12, interactive=False, placeholder="Translation will appear here")
+            translated_text = gr.Textbox(
+                label="Translation Result", 
+                lines=6, 
+                max_lines=12, 
+                interactive=False, 
+                placeholder="Translation will appear here"
+            )
 
     with gr.Row():
         with gr.Column():
@@ -168,15 +210,30 @@ with gr.Blocks(
                 inputs=[source_text, source_lang, target_lang],
             )
 
+    # Event handlers
     def swap_languages(src, tgt):
         return tgt, src
     
     def clear_all():
         return "", ""
 
-    swap_btn.click(fn=swap_languages, inputs=[source_lang, target_lang], outputs=[source_lang, target_lang])
-    clear_btn.click(fn=clear_all, outputs=[source_text, translated_text])
-    translate_btn.click(fn=translate_text, inputs=[source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens], outputs=[translated_text])
+    swap_btn.click(
+        fn=swap_languages, 
+        inputs=[source_lang, target_lang], 
+        outputs=[source_lang, target_lang]
+    )
+    
+    clear_btn.click(
+        fn=clear_all, 
+        outputs=[source_text, translated_text]
+    )
+    
+    translate_btn.click(
+        fn=translate_text, 
+        inputs=[source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens], 
+        outputs=[translated_text]
+    )
+    
     demo.load(fn=init_translation_model)
 
 if __name__ == "__main__":