Update app.py

app.py

@@ -1,16 +1,17 @@
-graidio
 import os
 import gradio as gr
 import pandas as pd
 import comtradeapicall
 from huggingface_hub import InferenceClient
+from deep_translator import GoogleTranslator
 
-# کلید COMTRADE (فعلاً برای preview نیازی به آن نداریم)
+# کلید COMTRADE
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 # توکن Hugging Face
 hf_token = os.getenv("HF_API_TOKEN")
 
 client = InferenceClient(token=hf_token)
+translator = GoogleTranslator(source='en', target='fa')
 
 def get_importers(hs_code: str, year: str, month: str):
     period = f"{year}{int(month):02d}"
@@ -22,14 +23,14 @@ def get_importers(hs_code: str, year: str, month: str):
         maxRecords=500, includeDesc=True
     )
     if df is None or df.empty:
-        return pd.DataFrame(columns=["کد کشور","نام کشور","ارزش CIF"])
+        return pd.DataFrame(columns=["کد کشور", "نام کشور", "ارزش CIF"])
     df = df[df['cifvalue'] > 0]
     result = (
-        df.groupby(["reporterCode","reporterDesc"], as_index=False)
-          .agg({"cifvalue":"sum"})
+        df.groupby(["reporterCode", "reporterDesc"], as_index=False)
+          .agg({"cifvalue": "sum"})
           .sort_values("cifvalue", ascending=False)
     )
-    result.columns = ["کد کشور","نام کشور","ارزش CIF"]
+    result.columns = ["کد کشور", "نام کشور", "ارزش CIF"]
     return result
 
 def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
@@ -38,18 +39,31 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     table_str = table_data.to_string(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = (
-        f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
+        f"The following table shows countries that imported a product with HS code {hs_code} during the period {period}:\n"
         f"{table_str}\n\n"
-        "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
+        f"Please provide a detailed and comprehensive analysis in two paragraphs. The first paragraph should discuss market opportunities, potential demand, and specific cultural or economic factors influencing the demand for this product in these countries. The second paragraph should offer actionable strategic recommendations for exporters, including detailed trade strategies, risk management techniques, and steps to establish local partnerships."
     )
-    # فراخوانی text_generation
-    outputs = client.text_generation(
-        model="google/gemma-2b",
-        inputs=prompt,
-        parameters={"max_new_tokens":256}
-    )
-    # خروجی اولین شات
-    return outputs[0]["generated_text"]
+    print("پرامپت ساخته‌شده:")
+    print(prompt)
+    try:
+        print("در حال فراخوانی مدل mistralai/Mixtral-8x7B-Instruct-v0.1...")
+        outputs = client.text_generation(
+            prompt=prompt,
+            model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+            max_new_tokens=1024  # افزایش برای تکمیل جملات
+        )
+        print("خروجی مدل دریافت شد (به انگلیسی):")
+        print(outputs)
+
+        # ترجمه خروجی به فارسی
+        translated_outputs = translator.translate(outputs)
+        print("خروجی ترجمه‌شده به فارسی:")
+        print(translated_outputs)
+        return translated_outputs
+    except Exception as e:
+        error_msg = f"خطا در تولید مشاوره: {str(e)}"
+        print(error_msg)
+        return error_msg
 
 current_year = pd.Timestamp.now().year
 years = [str(y) for y in range(2000, current_year+1)]
@@ -58,13 +72,13 @@ months = [str(m) for m in range(1, 13)]
 with gr.Blocks() as demo:
     gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
     with gr.Row():
-        inp_hs    = gr.Textbox(label="HS Code")
-        inp_year  = gr.Dropdown(choices=years, label="سال", value=str(current_year))
+        inp_hs = gr.Textbox(label="HS Code")
+        inp_year = gr.Dropdown(choices=years, label="سال", value=str(current_year))
         inp_month = gr.Dropdown(choices=months, label="ماه", value=str(pd.Timestamp.now().month))
     btn_show = gr.Button("نمایش اطلاعات")
     out_table = gr.Dataframe(
-        headers=["کد کشور","نام کشور","ارزش CIF"],
-        datatype=["number","text","number"],
+        headers=["کد کشور", "نام کشور", "ارزش CIF"],
+        datatype=["number", "text", "number"],
         interactive=True,
     )
     btn_show.click(get_importers, [inp_hs, inp_year, inp_month], out_table)
@@ -79,4 +93,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()