Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import os
 import pandas as pd
 import gradio as gr
@@ -6,6 +7,7 @@ from huggingface_hub import InferenceClient
 from deep_translator import GoogleTranslator
 import spaces  # برای مدیریت GPU کرایه‌ای
 
+# --- بارگذاری HS DATA از CSV گیت‌هاب ---
 HS_CSV_URL = (
     "https://raw.githubusercontent.com/"
     "datasets/harmonized-system/master/data/harmonized-system.csv"
@@ -17,6 +19,7 @@ def get_product_name(hs_code: str) -> str:
     row = hs_df[hs_df["hscode"] == code4]
     return row.iloc[0]["description"] if not row.empty else "–"
 
+# --- تابع دریافت واردات و پردازش ستون‌ها ---
 def get_importers(hs_code: str, year: str, month: str):
     product_name = get_product_name(hs_code)
     period = f"{year}{int(month):02d}"
@@ -30,24 +33,19 @@ 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):
         return product_name, df
 
     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
 
+# --- تابع تولید مشاوره تخصصی با GPU کرایه‌ای ---
 hf_token = os.getenv("HF_API_TOKEN")
 client = InferenceClient(token=hf_token)
 translator = GoogleTranslator(source='en', target='fa')
@@ -76,8 +74,19 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     except Exception as e:
         return f"خطا در تولید مشاوره: {e}"
 
-with gr.Blocks() as demo:
-    gr.Markdown("## تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی")
+# --- رابط کاربری Gradio با تنظیمات UI ---
+with gr.Blocks(css="""
+  /* رنگ پس‌زمینه سفید و متن سیاه */
+  body, .gradio-container { background-color: white !important; color: black !important; }
+  /* پنهان کردن فوتر و لینک‌های Gradio */
+  footer, .gradio-info { display: none !important; }
+""") as demo:
+    # عنوان سفارشی راست‌چین
+    gr.Markdown(
+        "<div dir='rtl' style='text-align: right; font-family: IRANSans;'>"
+        "<h2>هوش مصنوعی مشاوره صادراتی با HS Code محصول &ndash; ساخته شده توسط Diginoron</h2>"
+        "</div>"
+    )
 
     with gr.Row():
         inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
@@ -85,9 +94,10 @@ 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",
+        headers=['کد کشور', 'نام کشور', 'ارزش CIF'],
+        datatype=["number", "text", "number"],
         interactive=True
     )
 
@@ -98,7 +108,7 @@ with gr.Blocks() as demo:
     )
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
-    out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)
+    out_advice = gr.Textbox(label="<div dir='rtl' style='text-align: right; font-family: IRANSans;'>مشاوره تخصصی</div>", lines=8)
 
     btn_advice.click(
         fn=provide_advice,
@@ -107,4 +117,9 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        share=True,
+        show_api=False,
+        show_tips=False,
+        show_share=False
+    )