Update app.py

app.py

@@ -33,14 +33,24 @@ def get_importers(hs_code: str, year: str, month: str):
     if df is None or df.empty:
         return product_name, pd.DataFrame()
 
-    # شناسایی ستون‌های مورد نیاز
-    code_col = next((c for c in df.columns if 'code' in c.lower()), None)
-    title_col = next((c for c in df.columns if 'title' in c.lower()), None)
-    value_col = next((c for c in df.columns if 'value' in c.lower()), None)
+    # شناسایی ستون‌های مورد نیاز (کد کشور، نام کشور، ارزش)
+    # ابتدا سعی در استفاده از ستون‌های استاندارد
+    std_map = {
+        'کد کشور': 'ptCode',
+        'نام کشور': 'ptTitle',
+        'ارزش CIF': 'TradeValue'
+    }
+    code_col = std_map['کد کشور'] if 'ptCode' in df.columns else next((c for c in df.columns if 'code' in c.lower()), None)
+    title_col= std_map['نام کشور'] if 'ptTitle' in df.columns else next((c for c in df.columns if 'title' in c.lower()), None)
+    value_col= std_map['ارزش CIF'] if 'TradeValue' in df.columns else next((c for c in df.columns if 'value' in c.lower()), None)
+
     if not (code_col and title_col and value_col):
+        # اگر نتوانست ستون‌ها را شناسایی کند، برگرداندن DataFrame خام
         return product_name, df
 
+    # محدودسازی به 10 کشور برتر بر اساس ستون value_col
     df_sorted = df.sort_values(value_col, ascending=False).head(10)
+
     out = df_sorted[[code_col, title_col, value_col]]
     out.columns = ['کد کشور', 'نام کشور', 'ارزش CIF']
     return product_name, out
@@ -53,42 +63,31 @@ translator = GoogleTranslator(source='en', target='fa')
 @spaces.GPU
 def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
     if table_data is None or table_data.empty:
-        return "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">ابتدا نمایش داده‌های واردات را انجام دهید.</div>"
+        return "ابتدا نمایش داده‌های واردات را انجام دهید."
 
+    # محدودسازی تعداد ردیف‌های ورودی به 10 (در صورت بیشتر)
     df_limited = table_data.head(10)
     table_str = df_limited.to_string(index=False)
     period = f"{year}/{int(month):02d}"
-    prompt = f"""
-لطفاً یک گزارش تحلیلی به زبان فارسی و با چینش راست‌به‌چپ تولید کن، شامل دو بخش:
-1. جدول خلاصه (Markdown) 10 کشور برتر با ستون‌های رتبه، نام کشور، ارزش CIF و سه تأمین‌کننده برتر.
-2. نکات کلیدی و پیشنهادات به‌صورت بولت‌پوینت با تراز راست.
-
-- متن خروجی را درون یک <div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\"> قرار بده.
-
-داده‌ها:
-{table_str}
-""" 10 کشور برتر با ستون‌های رتبه، نام کشور، ارزش CIF و سه تأمین‌کننده برتر."
-        "\n2. نکات کلیدی و پیشنهادات به‌صورت بولت‌پوینت با تراز راست."
-        "\n
-- متن خروجی را درون یک <div dir=\\\"rtl\\\" style=\\\"text-align: right; font-family: IRANSans;\\\"> قرار بده."
-        f"\n
-داده‌ها:\n{table_str}"
+    prompt = (
+        f"The following table shows the top {len(df_limited)} countries by CIF value importing HS code {hs_code} during {period}:\n"
+        f"{table_str}\n\n"
+        "Please provide a detailed and comprehensive analysis of market trends, risks, "
+        "and opportunities for a new exporter entering this market."
     )
     try:
         outputs = client.text_generation(
             prompt=prompt,
             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-            max_new_tokens=2048
+            max_new_tokens=1024
         )
-        persian = translator.translate(outputs)
-        # قرار دادن در کانتینر HTML راست‌چین
-        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">{persian}</div>"
+        return translator.translate(outputs)
     except Exception as e:
-        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">خطا در تولید مشاوره: {e}</div>"
+        return f"خطا در تولید مشاوره: {e}"
 
 # --- رابط کاربری Gradio ---
 with gr.Blocks() as demo:
-    gr.Markdown("<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\"><h2>تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی</h2></div>", elem_id="title")
+    gr.Markdown("## تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی")
 
     with gr.Row():
         inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
@@ -96,7 +95,7 @@ with gr.Blocks() as demo:
         inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
 
     btn_show = gr.Button("نمایش داده‌های واردات")
-    out_name = gr.Markdown(label="**<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">نام محصول</div>")
+    out_name = gr.Markdown(label="**نام محصول**")
     out_table = gr.Dataframe(
         datatype="pandas",
         interactive=True
@@ -109,7 +108,7 @@ with gr.Blocks() as demo:
     )
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
-    out_advice = gr.HTML()
+    out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)
 
     btn_advice.click(
         fn=provide_advice,
@@ -118,4 +117,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()