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,41 +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; color: black; background: white;\">ابتدا نمایش داده‌های واردات را انجام دهید.</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) ۱۰ کشور برتر با ستون‌های «رتبه»، «نام کشور»، «ارزش CIF» و «سه تأمین‌کننده برتر».\n
-2. **نکات کلیدی و پیشنهادات** به‌صورت بولت‌پوینت با تراز راست.
-
-
-**داده‌ها:**
-{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)
-        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">{persian}</div>"
+        return translator.translate(outputs)
     except Exception as e:
-        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">خطا در تولید مشاوره: {e}</div>"
+        return f"خطا در تولید مشاوره: {e}"
 
 # --- رابط کاربری Gradio ---
-with gr.Blocks(css="""
-body, .gradio-container {background-color: white; color: black;}
-""") as demo:
-    gr.Markdown(
-        "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">"
-        "<h2>هوش مصنوعی مشاوره صادراتی با HS Code محصول – ساخته شده توسط Diginoron</h2>"
-        "</div>"
-    )
+with gr.Blocks() as demo:
+    gr.Markdown("## تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی")
 
     with gr.Row():
         inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
@@ -95,14 +95,9 @@ body, .gradio-container {background-color: white; color: black;}
         inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
 
     btn_show = gr.Button("نمایش داده‌های واردات")
-    out_name = gr.Markdown(
-        "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">"
-        "**نام محصول**"
-        "</div>"
-    )
+    out_name = gr.Markdown(label="**نام محصول**")
     out_table = gr.Dataframe(
-        headers=["کد کشور", "نام کشور", "ارزش CIF"],
-        datatype=["text", "text", "text"],
+        datatype="pandas",
         interactive=True
     )
 
@@ -113,7 +108,7 @@ body, .gradio-container {background-color: white; color: black;}
     )
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
-    out_advice = gr.HTML()
+    out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)
 
     btn_advice.click(
         fn=provide_advice,