Update app.py

app.py

@@ -5,6 +5,8 @@ subprocess.run(
     env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"},
     shell=True
 )
+subprocess.run('pip install googletrans==4.0.0-rc1', shell=True)
+
 import os
 import re
 import time
@@ -12,6 +14,7 @@ import torch
 import spaces
 import gradio as gr
 from threading import Thread
+from googletrans import Translator
 from transformers import (
     AutoModelForCausalLM, 
     AutoTokenizer, 
@@ -28,7 +31,6 @@ Writing Style:
 
 1. Grammar Accuracy: Always ensure translations are grammatically correct.
 
-
 2. Contextual Suitability: Tailor translations to the context and audience:
 
 Use concise, clear sentences for medical and legal content.
@@ -37,12 +39,8 @@ Employ engaging, imaginative language for marketing material.
 
 Preserve formality and eloquence for legal content.
 
-
-
 3. Proper Structure: Respect Arabic sentence structures and avoid direct replication of source language grammar. Favor "الجملة الفعلية" unless "الجملة الاسمية" is more suitable (e.g., for headlines or disclaimers).
 
-
-
 Style Choices:
 
 Use diacritics only when necessary for clarity.
@@ -53,10 +51,8 @@ Transliterate names and drug names unless an Arabic equivalent exists.
 
 Translate program, department, and agency names when beneficial.
 
-
 Use Arabic numerals and ensure proper handling of units, addresses, and references.
 
-
 Punctuation:
 
 Apply Arabic punctuation rules, ensuring proper readability.
@@ -65,8 +61,6 @@ Use the Arabic comma (،) and semicolon (؛) as per conventions.
 
 Avoid excessive use of quotation marks and ensure logical placement of colons (:).
 
-
-
 Common Mistakes to Avoid:
 
 Avoid translating "is" as "is considered" unless contextually appropriate.
@@ -77,7 +71,6 @@ Minimize repetitive structures; leverage pronouns where applicable.
 
 Avoid overuse of constructions like "(قام + الفعل)" and "الخاص بـ."
 
-
 Specific Terminology:
 
 For legal translations, maintain formal tone and ensure accuracy in terminology.
@@ -86,14 +79,12 @@ For medical translations, simplify technical terms for lay audiences but retain
 
 For marketing translations, prioritize creativity over literal translation, aligning with the core message.
 
-
 Formatting Guidelines:
 
 Consistently follow Arabic typographic standards.
 
 Preserve the format of critical data (e.g., dates, measurements, and legal citations).
 
-
 When in doubt, prioritize clarity, consistency, and alignment with the target audience's needs. Always reconcile project-specific instructions with these guidelines, giving precedence to client requirements when conflicts arise.
 """
 # UI Configuration
@@ -271,6 +262,14 @@ def chat_response(
             
             yield history, chat_display
 
+    # Translate the final response to Arabic
+    translator = Translator()
+    translated_text = translator.translate(buffer, src='en', dest='ar').text
+    history[-1][1] = translated_text
+    chat_display = format_chat_history(history)
+    
+    yield history, chat_display
+
 def process_example(example: str) -> tuple:
     """Process example query and return empty history and updated display"""
     return [], f"User: {example}\n\n"