Update app.py

app.py

@@ -34,6 +34,7 @@ def get_importers(hs_code: str, year: str, month: str):
         return product_name, pd.DataFrame()
 
     # شناسایی ستون‌های مورد نیاز (کد کشور، نام کشور، ارزش)
+    # ابتدا سعی در استفاده از ستون‌های استاندارد
     std_map = {
         'کد کشور': 'ptCode',
         'نام کشور': 'ptTitle',
@@ -44,6 +45,7 @@ def get_importers(hs_code: str, year: str, month: str):
     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
@@ -83,27 +85,9 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     except Exception as e:
         return f"خطا در تولید مشاوره: {e}"
 
-# --- رابط کاربری Gradio با تنظیمات UI ---
-with gr.Blocks(css="""
-  /* رنگ پس‌زمینه سفید و متن سیاه */
-  body, .gradio-container { background-color: white !important; color: black !important; }
-  /* پنهان کردن فوتر و لینک‌های Gradio */
-  footer, .gradio-info { display: none !important; }
-  /* افزودن استایل برای جدول */
-  .gradio-table { border: 1px solid #ddd; border-radius: 8px; }
-  /* استایل برای دکمه‌ها */
-  .gr-button { background-color: #4CAF50; color: white; font-size: 16px; }
-  .gr-button:hover { background-color: #45a049; }
-""") as demo:
-    # افزودن لوگو
-    gr.Image("https://cdn-ilalfgl.nitrocdn.com/CdpVmnTBcinnSxXeznJeoSCevtRYCUap/assets/images/optimized/rev-d50e51b/diginoron.com/wp-content/uploads/2024/06/diginoron-e1718654683899.png", elem_id="logo", interactive=False)
-
-    # عنوان سفارشی راست‌چین
-    gr.Markdown(
-        "<div dir='rtl' style='text-align: right; font-family: IRANSans;'>"
-        "<h2>هوش مصنوعی مشاوره صادراتی با HS Code محصول – ساخته شده توسط Diginoron</h2>"
-        "</div>"
-    )
+# --- رابط کاربری Gradio ---
+with gr.Blocks() as demo:
+    gr.Markdown("## تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی")
 
     with gr.Row():
         inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
@@ -133,4 +117,4 @@ with gr.Blocks(css="""
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()