Update app.py

app.py

@@ -18,10 +18,10 @@ model = PeftModel.from_pretrained(base_model, adapter_model_name)
 
 def translate(text, source_lang, target_lang):
     if not text.strip():
-        return "Please enter text to translate", "0 characters"
+        return ""
     
     if source_lang == target_lang:
-        return "Source and target languages cannot be the same", f"{len(text)} characters"
+        return text
     
     lang_map = {
         "English": "eng_Latn",
@@ -46,78 +46,220 @@ def translate(text, source_lang, target_lang):
     )
     
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    char_info = f"{len(text)} → {len(result)} characters"
+    return result
+
+def swap_languages(src, tgt, input_txt, output_txt):
+    return tgt, src, output_txt, input_txt
+
+def load_file(file):
+    if file is None:
+        return ""
     
-    return result, char_info
+    try:
+        with open(file.name, 'r', encoding='utf-8') as f:
+            content = f.read()
+        return content
+    except:
+        try:
+            with open(file.name, 'r', encoding='latin-1') as f:
+                content = f.read()
+            return content
+        except Exception as e:
+            return f"Error reading file: {str(e)}"
 
-def swap_languages(src, tgt):
-    return tgt, src
+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
 
-def clear_all():
-    return "", "", "0 characters"
+custom_css = """
+.gradio-container {
+    max-width: 1400px !important;
+    margin: auto !important;
+}
+.lang-selector {
+    border-radius: 12px !important;
+    border: 1px solid #e0e0e0 !important;
+    font-size: 16px !important;
+}
+.swap-btn {
+    min-width: 48px !important;
+    height: 48px !important;
+    border-radius: 50% !important;
+    background: white !important;
+    border: 1px solid #dadce0 !important;
+    font-size: 20px !important;
+}
+.text-area {
+    border: none !important;
+    border-radius: 8px !important;
+    font-size: 16px !important;
+    line-height: 1.6 !important;
+}
+.translate-container {
+    background: white !important;
+    border-radius: 8px !important;
+    box-shadow: 0 1px 3px rgba(0,0,0,0.1) !important;
+}
+.example-box {
+    border-left: 3px solid #1a73e8 !important;
+    padding-left: 12px !important;
+    margin: 8px 0 !important;
+    background: #f8f9fa !important;
+    border-radius: 4px !important;
+}
+"""
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# Oil & Gas Professional Translation")
-    gr.Markdown("English ↔ Norwegian translation specialized for petroleum industry")
-    
-    with gr.Row():
-        source_lang = gr.Dropdown(
-            choices=["English", "Norwegian"],
-            label="Source Language",
-            value="English"
-        )
-        swap_btn = gr.Button("⇄ Swap", size="sm")
-        target_lang = gr.Dropdown(
-            choices=["English", "Norwegian"],
-            label="Target Language",
-            value="Norwegian"
-        )
+with gr.Blocks(css=custom_css, theme=gr.themes.Default()) as demo:
     
-    with gr.Row():
-        with gr.Column():
+    with gr.Row(elem_classes="translate-container"):
+        with gr.Column(scale=1):
+            with gr.Row():
+                source_lang = gr.Dropdown(
+                    choices=["English", "Norwegian"],
+                    value="English",
+                    show_label=False,
+                    container=False,
+                    elem_classes="lang-selector"
+                )
+            
             input_text = gr.Textbox(
-                label="Input Text",
-                placeholder="Enter text to translate",
-                lines=8
+                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.Column():
+        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=1):
+            with gr.Row():
+                target_lang = gr.Dropdown(
+                    choices=["English", "Norwegian"],
+                    value="Norwegian",
+                    show_label=False,
+                    container=False,
+                    elem_classes="lang-selector"
+                )
+            
             output_text = gr.Textbox(
-                label="Translation",
-                lines=8
+                placeholder="Translation",
+                show_label=False,
+                lines=12,
+                container=False,
+                elem_classes="text-area",
+                interactive=False
+            )
+            
+            download_btn = gr.File(
+                label="Download translation",
+                visible=False
             )
-    
-    char_count = gr.Textbox(label="Statistics", value="0 characters", interactive=False)
     
     with gr.Row():
-        translate_btn = gr.Button("Translate", variant="primary", size="lg")
-        clear_btn = gr.Button("Clear All", size="lg")
+        gr.Markdown(
+            "<center style='color: #666; font-size: 14px; margin-top: 20px;'>"
+            "Oil & Gas specialized translation • Powered by NLLB-200 + LoRA"
+            "</center>"
+        )
     
-    gr.Examples(
-        examples=[
-            ["The drilling operation encountered high pressure at 3000 meters depth", "English", "Norwegian"],
-            ["Reservoaret viser god permeabilitet og porøsitet", "Norwegian", "English"],
-            ["Wellhead pressure monitoring indicates stable production", "English", "Norwegian"]
-        ],
+    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>
+                    """
+                )
+    
+    input_text.change(
+        fn=translate,
+        inputs=[input_text, source_lang, target_lang],
+        outputs=output_text
+    )
+    
+    source_lang.change(
+        fn=translate,
         inputs=[input_text, source_lang, target_lang],
-        label="Example Translations"
+        outputs=output_text
     )
     
-    translate_btn.click(
+    target_lang.change(
         fn=translate,
         inputs=[input_text, source_lang, target_lang],
-        outputs=[output_text, char_count]
+        outputs=output_text
     )
     
     swap_btn.click(
         fn=swap_languages,
-        inputs=[source_lang, target_lang],
-        outputs=[source_lang, target_lang]
+        inputs=[source_lang, target_lang, input_text, output_text],
+        outputs=[source_lang, target_lang, input_text, output_text]
     )
     
-    clear_btn.click(
-        fn=clear_all,
-        outputs=[input_text, output_text, char_count]
+    file_input.change(
+        fn=load_file,
+        inputs=file_input,
+        outputs=input_text
     )
 
 demo.launch()