Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import pandas as pd
 import comtradeapicall
 from huggingface_hub import InferenceClient
+from googletrans import Translator  # اضافه کردن کتابخانه ترجمه
 
 # کلید COMTRADE
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
@@ -10,6 +11,7 @@ subscription_key = os.getenv("COMTRADE_API_KEY", "")
 hf_token = os.getenv("HF_API_TOKEN")
 
 client = InferenceClient(token=hf_token)
+translator = Translator()  # ایجاد یک نمونه از مترجم
 
 def get_importers(hs_code: str, year: str, month: str):
     period = f"{year}{int(month):02d}"
@@ -48,11 +50,16 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         outputs = client.text_generation(
             prompt=prompt,
             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-            max_new_tokens=512  # افزایش طول خروجی
+            max_new_tokens=512
         )
-        print("خروجی مدل دریافت شد:")
+        print("خروجی مدل دریافت شد (به انگلیسی):")
         print(outputs)
-        return outputs
+
+        # ترجمه خروجی به فارسی
+        translated_outputs = translator.translate(outputs, src='en', dest='fa').text
+        print("خروجی ترجمه‌شده به فارسی:")
+        print(translated_outputs)
+        return translated_outputs
     except Exception as e:
         error_msg = f"خطا در تولید مشاوره: {str(e)}"
         print(error_msg)