Update app.py

app.py

@@ -5,31 +5,23 @@ from huggingface_hub import hf_hub_download, snapshot_download, login
 import sys
 import gc
 import base64
+import torch
 
 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 from private repo
 try:
-    generate_file = hf_hub_download(
-        repo_id=repo_id,
-        filename="gen1.py",
-        token=os.environ["HF_TOKEN"]
-    )
+    generate_file = hf_hub_download(repo_id=repo_id, filename="gen1.py", token=os.environ["HF_TOKEN"])
     os.system(f"cp {generate_file} ./gen1.py")
 except Exception as e:
     print(f"Error downloading files: {e}")
 
-# Import the generation wrapper
 from gen1 import setup_translation, translate_text as gen_translate
 
-# Logo and favicon setup
 LOGO_PATH = "static/logo.png"
 if os.path.isfile(LOGO_PATH):
     with open(LOGO_PATH, "rb") as f:
@@ -41,29 +33,18 @@ else:
     FAVICON_HTML = ''
 
 def init_translation_model():
-    """Initialize the translation model from private repo"""
-    success = setup_translation(
-        repo_id=os.environ["REPO_ID"],
-        token=os.environ["HF_TOKEN"]
-    )
+    success = setup_translation(repo_id=os.environ["REPO_ID"], token=os.environ["HF_TOKEN"])
     if success:
         print("Model loaded successfully!")
     else:
         print("Failed to load model")
 
 def translate_text(source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens):
-    """Handle translation requests"""
-    return gen_translate(
-        source_text, source_lang, target_lang, 
-        temperature, top_k, repetition_penalty, max_tokens
-    )
+    return gen_translate(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 ""
+# 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,
@@ -82,14 +63,8 @@ body, .gradio-container {
 }
 """
 
-combined_css = custom_css + theme_lock_css
-
-# Theme configuration
-locked_theme = gr.themes.Monochrome(
-    primary_hue="blue",
-    secondary_hue="slate",
-    neutral_hue="slate"
-).set(
+# combined_css = theme_lock_css
+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",
@@ -100,10 +75,9 @@ locked_theme = gr.themes.Monochrome(
     body_text_color="#e6eef8"
 )
 
-# Create Gradio interface
 with gr.Blocks(
     title="Madhuram Translation - MaruthLabs",
-    css=combined_css,
+    css=theme_lock_css,
     theme=locked_theme,
     js=f"""
     function() {{
@@ -128,7 +102,6 @@ with gr.Blocks(
     """
 ) as demo:
 
-    # Header section
     with gr.Row(elem_classes="main-header"):
         with gr.Column():
             gr.HTML(f"""
@@ -147,9 +120,7 @@ with gr.Blocks(
             </div>
             """)
 
-    # Main interface
     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():
@@ -163,7 +134,6 @@ with gr.Blocks(
                 repetition_penalty = gr.Slider(1.0, 2.0, 1.2, step=0.1, label="Repetition Penalty")
                 max_tokens = gr.Slider(100, 2000, 400, step=50, label="Max Tokens")
 
-        # Translation interface
         with gr.Column(scale=2, elem_classes="translation-card"):
             gr.Markdown("## Translation Interface")
             source_text = gr.Textbox(
@@ -183,7 +153,6 @@ with gr.Blocks(
                 placeholder="Translation will appear here"
             )
 
-    # Examples section
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Quick Examples")
@@ -198,32 +167,15 @@ with gr.Blocks(
                 inputs=[source_text, source_lang, target_lang],
             )
 
-    # Event handlers
     def swap_languages(src, tgt):
         return tgt, src
     
     def clear_all():
         return "", ""
 
-    # Connect event handlers
-    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]
-    )
-    
-    # Load model when demo starts
+    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__":