Update app.py

app.py

@@ -33,24 +33,14 @@ def get_importers(hs_code: str, year: str, month: str):
     if df is None or df.empty:
         return product_name, pd.DataFrame()
 
-    # شناسایی ستون‌های مورد نیاز (کد کشور، نام کشور، ارزش)
-    # ابتدا سعی در استفاده از ستون‌های استاندارد
-    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)
-
+    # شناسایی ستون‌های مورد نیاز
+    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)
     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
@@ -63,31 +53,42 @@ 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 "ابتدا نمایش داده‌های واردات را انجام دهید."
+        return "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">ابتدا نمایش داده‌های واردات را انجام دهید.</div>"
 
-    # محدودسازی تعداد ردیف‌های ورودی به 10 (در صورت بیشتر)
     df_limited = table_data.head(10)
     table_str = df_limited.to_string(index=False)
     period = f"{year}/{int(month):02d}"
-    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."
+    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}"
     )
     try:
         outputs = client.text_generation(
             prompt=prompt,
             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-            max_new_tokens=1024
+            max_new_tokens=2048
         )
-        return translator.translate(outputs)
+        persian = translator.translate(outputs)
+        # قرار دادن در کانتینر HTML راست‌چین
+        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">{persian}</div>"
     except Exception as e:
-        return f"خطا در تولید مشاوره: {e}"
+        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">خطا در تولید مشاوره: {e}</div>"
 
 # --- رابط کاربری Gradio ---
 with gr.Blocks() as demo:
-    gr.Markdown("## تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی")
+    gr.Markdown("<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\"><h2>تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی</h2></div>", elem_id="title")
 
     with gr.Row():
         inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
@@ -95,7 +96,7 @@ with gr.Blocks() as demo:
         inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
 
     btn_show = gr.Button("نمایش داده‌های واردات")
-    out_name = gr.Markdown(label="**نام محصول**")
+    out_name = gr.Markdown(label="**<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">نام محصول</div>")
     out_table = gr.Dataframe(
         datatype="pandas",
         interactive=True
@@ -108,7 +109,7 @@ with gr.Blocks() as demo:
     )
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
-    out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)
+    out_advice = gr.HTML()
 
     btn_advice.click(
         fn=provide_advice,
@@ -117,4 +118,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()