Update app.py

app.py

@@ -28,25 +28,35 @@ def translate(text, source_lang, target_lang):
         "Norwegian": "nob_Latn"
     }
     
-    inputs = tokenizer(
-        text,
-        return_tensors="pt",
-        truncation=True,
-        max_length=512
-    )
-    
-    if hasattr(model, 'device'):
-        inputs = {k: v.to(model.device) for k, v in inputs.items()}
+    sentences = text.split('\n')
+    translated_sentences = []
     
-    outputs = model.generate(
-        **inputs,
-        forced_bos_token_id=tokenizer.convert_tokens_to_ids(lang_map[target_lang]),
-        max_length=512,
-        num_beams=5
-    )
+    for sentence in sentences:
+        if not sentence.strip():
+            translated_sentences.append("")
+            continue
+        
+        inputs = tokenizer(
+            sentence,
+            return_tensors="pt",
+            truncation=True,
+            max_length=512
+        )
+        
+        if hasattr(model, 'device'):
+            inputs = {k: v.to(model.device) for k, v in inputs.items()}
+        
+        outputs = model.generate(
+            **inputs,
+            forced_bos_token_id=tokenizer.convert_tokens_to_ids(lang_map[target_lang]),
+            max_length=512,
+            num_beams=5
+        )
+        
+        result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        translated_sentences.append(result)
     
-    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return result
+    return '\n'.join(translated_sentences)
 
 def swap_languages(src, tgt, input_txt, output_txt):
     return tgt, src, output_txt, input_txt
@@ -67,170 +77,150 @@ def load_file(file):
         except Exception as e:
             return f"Error reading file: {str(e)}"
 
-def translate_file(file, source_lang, target_lang):
-    content = load_file(file)
-    if content and not content.startswith("Error"):
-        return translate(content, source_lang, target_lang)
-    return content
-
 custom_css = """
 .gradio-container {
-    max-width: 1400px !important;
-    margin: auto !important;
+    max-width: 1100px !important;
+    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif !important;
 }
-.lang-selector {
-    border-radius: 12px !important;
-    border: 1px solid #e0e0e0 !important;
-    font-size: 16px !important;
+.main-container {
+    background: #f6f7f8 !important;
+    padding: 0 !important;
+    border-radius: 0 !important;
 }
-.swap-btn {
-    min-width: 48px !important;
-    height: 48px !important;
-    border-radius: 50% !important;
+.translate-box {
     background: white !important;
-    border: 1px solid #dadce0 !important;
-    font-size: 20px !important;
+    border-radius: 5px !important;
+    padding: 0 !important;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.08) !important;
+    margin: 20px 0 !important;
+}
+.lang-header {
+    padding: 16px 20px !important;
+    border-bottom: 1px solid #e8eaed !important;
+    background: #fafafa !important;
+}
+.lang-selector {
+    border: none !important;
+    background: transparent !important;
+    font-size: 15px !important;
+    font-weight: 500 !important;
+    color: #333 !important;
 }
-.text-area {
+.text-area textarea {
     border: none !important;
-    border-radius: 8px !important;
-    font-size: 16px !important;
-    line-height: 1.6 !important;
+    font-size: 17px !important;
+    line-height: 1.7 !important;
+    padding: 20px !important;
+    min-height: 200px !important;
+}
+.swap-container {
+    display: flex !important;
+    align-items: center !important;
+    justify-content: center !important;
+    padding: 20px 0 !important;
 }
-.translate-container {
+.swap-btn {
+    width: 44px !important;
+    height: 44px !important;
+    min-width: 44px !important;
+    border-radius: 50% !important;
     background: white !important;
-    border-radius: 8px !important;
+    border: 1px solid #d1d5db !important;
     box-shadow: 0 1px 3px rgba(0,0,0,0.1) !important;
+    font-size: 18px !important;
+    color: #0f6fff !important;
+    cursor: pointer !important;
 }
-.example-box {
-    border-left: 3px solid #1a73e8 !important;
-    padding-left: 12px !important;
-    margin: 8px 0 !important;
+.swap-btn:hover {
     background: #f8f9fa !important;
-    border-radius: 4px !important;
+    border-color: #0f6fff !important;
+}
+.footer-info {
+    text-align: center !important;
+    color: #999 !important;
+    font-size: 13px !important;
+    padding: 20px !important;
 }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Default()) as demo:
     
-    with gr.Row(elem_classes="translate-container"):
+    gr.HTML("<div style='height: 20px'></div>")
+    
+    with gr.Row():
         with gr.Column(scale=1):
-            with gr.Row():
-                source_lang = gr.Dropdown(
-                    choices=["English", "Norwegian"],
-                    value="English",
+            with gr.Group(elem_classes="translate-box"):
+                with gr.Row(elem_classes="lang-header"):
+                    source_lang = gr.Dropdown(
+                        choices=["English", "Norwegian"],
+                        value="English",
+                        show_label=False,
+                        container=False,
+                        elem_classes="lang-selector",
+                        scale=1
+                    )
+                
+                input_text = gr.Textbox(
+                    placeholder="Type to translate",
                     show_label=False,
+                    lines=8,
+                    max_lines=20,
                     container=False,
-                    elem_classes="lang-selector"
+                    elem_classes="text-area"
                 )
-            
-            input_text = gr.Textbox(
-                placeholder="Enter text",
-                show_label=False,
-                lines=12,
-                container=False,
-                elem_classes="text-area"
-            )
-            
-            file_input = gr.File(
-                label="Upload text file",
-                file_types=[".txt", ".md", ".doc"],
-                type="filepath"
-            )
+                
+                with gr.Row(visible=True):
+                    file_input = gr.File(
+                        label="",
+                        file_types=[".txt"],
+                        type="filepath",
+                        scale=1,
+                        container=False
+                    )
         
-        with gr.Column(scale=0, min_width=80):
-            gr.Markdown("<br><br>")
-            swap_btn = gr.Button("⇄", elem_classes="swap-btn", size="sm")
+        with gr.Column(scale=0, min_width=100):
+            with gr.Row(elem_classes="swap-container"):
+                swap_btn = gr.Button("⇄", elem_classes="swap-btn")
         
         with gr.Column(scale=1):
-            with gr.Row():
-                target_lang = gr.Dropdown(
-                    choices=["English", "Norwegian"],
-                    value="Norwegian",
+            with gr.Group(elem_classes="translate-box"):
+                with gr.Row(elem_classes="lang-header"):
+                    target_lang = gr.Dropdown(
+                        choices=["English", "Norwegian"],
+                        value="Norwegian",
+                        show_label=False,
+                        container=False,
+                        elem_classes="lang-selector",
+                        scale=1
+                    )
+                
+                output_text = gr.Textbox(
+                    placeholder="Translation",
                     show_label=False,
+                    lines=8,
+                    max_lines=20,
                     container=False,
-                    elem_classes="lang-selector"
+                    elem_classes="text-area",
+                    interactive=False
                 )
-            
-            output_text = gr.Textbox(
-                placeholder="Translation",
-                show_label=False,
-                lines=12,
-                container=False,
-                elem_classes="text-area",
-                interactive=False
-            )
-            
-            download_btn = gr.File(
-                label="Download translation",
-                visible=False
-            )
     
-    with gr.Row():
-        gr.Markdown(
-            "<center style='color: #666; font-size: 14px; margin-top: 20px;'>"
-            "Oil & Gas specialized translation • Powered by NLLB-200 + LoRA"
-            "</center>"
-        )
+    gr.HTML(
+        "<div class='footer-info'>"
+        "Oil & Gas Translation • English ↔ Norwegian"
+        "</div>"
+    )
     
-    with gr.Accordion("Example Translations", open=False):
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown(
-                    """
-                    <div class='example-box'>
-                    <b>Drilling Report Example:</b><br><br>
-                    The drilling operation at well site A-15 encountered unexpected high-pressure zones at 3,247 meters depth. 
-                    The mud weight was increased from 1.65 to 1.82 specific gravity to maintain wellbore stability. 
-                    Pressure readings indicate a potential hydrocarbon-bearing formation with estimated reservoir pressure 
-                    of 4,850 psi. The drilling team implemented managed pressure drilling techniques to safely penetrate 
-                    the formation while maintaining kick tolerance within acceptable limits.
-                    </div>
-                    """
-                )
-            
-            with gr.Column():
-                gr.Markdown(
-                    """
-                    <div class='example-box'>
-                    <b>Reservoir Analysis Example:</b><br><br>
-                    Reservoaret viser utmerket permeabilitet på 250 millidarcy og porøsitet på 22 prosent basert på 
-                    kjerneanalyse. Formasjonsvannsmetningen er estimert til 35 prosent, noe som indikerer betydelig 
-                    hydrokarbonpotensial. Seismiske data bekrefter en strukturell felle med estimert areal på 12 
-                    kvadratkilometer. Produktivitetstester viser stabilisert oljeproduksjon på 3,400 fat per dag ved 
-                    optimaliseringstrykk på 2,100 psi.
-                    </div>
-                    """
-                )
-        
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown(
-                    """
-                    <div class='example-box'>
-                    <b>Safety Procedure Example:</b><br><br>
-                    All personnel must complete H2S safety training before entering the drilling site. Emergency response 
-                    equipment including breathing apparatus and wind direction indicators must be positioned at designated 
-                    muster points. Wellhead pressure monitoring systems shall be checked every 4 hours during critical 
-                    operations. In case of gas detection exceeding 10 ppm, immediate evacuation procedures must be initiated 
-                    according to the site emergency response plan.
-                    </div>
-                    """
-                )
-            
-            with gr.Column():
-                gr.Markdown(
-                    """
-                    <div class='example-box'>
-                    <b>Equipment Specification Example:</b><br><br>
-                    Undervannsproduktjonssystemet består av en vertikalt juletræ med maksimal arbeidstrykk på 10,000 psi 
-                    og temperaturbegrensning på 150 grader Celsius. Produksjonsrøret er 7-tommers krom-legering designet 
-                    for korrosive miljøer. Flytkontrollen opprettholdes ved bruk av subsea multifase meter og akustisk 
-                    sanddeteksjon. Systemet er utstyrt med redundante sikkerhetsfunksjoner inkludert automatiske 
-                    nedstengningsventiler og trykkovervåkingssystemer.
-                    </div>
-                    """
-                )
+    with gr.Accordion("Example Sentences", open=False):
+        gr.Examples(
+            examples=[
+                ["The drilling operation at well site A-15 encountered unexpected high-pressure zones at 3,247 meters depth, requiring immediate adjustment of mud weight from 1.65 to 1.82 specific gravity to maintain wellbore stability and prevent potential kicks.", "English", "Norwegian"],
+                ["Reservoaret viser utmerket permeabilitet på 250 millidarcy og porøsitet på 22 prosent basert på kjerneanalyse, noe som indikerer betydelig hydrokarbonpotensial med estimert oljemetning på 65 prosent.", "Norwegian", "English"],
+                ["The subsea production system consists of a vertical Christmas tree rated for 10,000 psi working pressure and 150 degrees Celsius temperature, equipped with redundant safety features including automatic shutdown valves and real-time pressure monitoring systems.", "English", "Norwegian"],
+                ["Seismiske data bekrefter tilstedeværelsen av en strukturell felle med estimert areal på 12 kvadratkilometer, og produktivitetstester viser stabilisert oljeproduksjon på 3,400 fat per dag ved optimaliseringstrykk på 2,100 psi.", "Norwegian", "English"],
+                ["Emergency response procedures require all personnel to complete H2S safety training before site access, with breathing apparatus and wind indicators positioned at designated muster points, and immediate evacuation protocols activated when gas detection exceeds 10 ppm concentration levels.", "English", "Norwegian"]
+            ],
+            inputs=[input_text, source_lang, target_lang],
+        )
     
     input_text.change(
         fn=translate,