Update app.py

app.py

@@ -3,11 +3,11 @@ import pandas as pd
 import comtradeapicall
 import os
 
+# کلید API (برای preview نیازی نیست)
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 proxy_url        = None
 
 def get_importers(hs_code: str, period: str):
-    # ۱. فراخوانی preview تا ۵۰۰ رکورد
     df = comtradeapicall.previewFinalData(
         typeCode='C', freqCode='M', clCode='HS', period=period,
         reporterCode=None, cmdCode=hs_code, flowCode='M',
@@ -17,28 +17,26 @@ def get_importers(hs_code: str, period: str):
         proxy_url=proxy_url
     )
     if df is None or df.empty:
-        # اگر خالی بود، یک DataFrame خالی برمی‌گردانیم
-        return pd.DataFrame(columns=["reporterCode","reporterDesc","cifvalue"])
+        return pd.DataFrame(columns=["کد کشور","نام کشور","ارزش CIF"])
     
-    # ۲. فیلتر و مرتب‌سازی
     df = df[df['cifvalue'] > 0]
     df = df.sort_values('cifvalue', ascending=False)
     
-    # ۳. ستون‌های مورد نظر
+    # فقط سه ستون اصلی
     result = df[["reporterCode","reporterDesc","cifvalue"]].copy()
+    # تغییر نام ستون‌ها به فارسی
+    result.columns = ["کد کشور", "نام کشور", "ارزش CIF"]
     result.reset_index(drop=True, inplace=True)
     return result
 
 with gr.Blocks() as demo:
     gr.Markdown("## جست‌وجوی واردکننده‌های یک HS-Code")
-    
     with gr.Row():
-        inp_hs     = gr.Textbox(label="HS Code (مثلا 010121)")
-        inp_period = gr.Textbox(label="دوره (YYYYMM، مثلا 202205)")
-    
+        inp_hs     = gr.Textbox(label="HS Code (مثلاً 010121)")
+        inp_period = gr.Textbox(label="دوره (YYYYMM، مثلاً 202205)")
     btn = gr.Button("نمایش کشورها")
     out = gr.Dataframe(
-        headers=["reporterCode","reporterDesc","cifvalue"],
+        headers=["کد کشور", "نام کشور", "ارزش CIF"],
         datatype=["number","text","number"],
         interactive=False,
     )