Update app.py

app.py

@@ -3,11 +3,11 @@ import pandas as pd
 import comtradeapicall
 import os
 
-# ۱) متغیرهای عمومی
-subscription_key = os.getenv("COMTRADE_API_KEY", "") 
+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,18 +17,31 @@ def get_importers(hs_code: str, period: str):
         proxy_url=proxy_url
     )
     if df is None or df.empty:
-        return "هیچ داده‌ای یافت نشد."
+        # اگر خالی بود، یک DataFrame خالی برمی‌گردانیم
+        return pd.DataFrame(columns=["reporterCode","reporterDesc","cifvalue"])
+    
+    # ۲. فیلتر و مرتب‌سازی
     df = df[df['cifvalue'] > 0]
     df = df.sort_values('cifvalue', ascending=False)
-    return df['reporterDesc'].tolist()
+    
+    # ۳. ستون‌های مورد نظر
+    result = df[["reporterCode","reporterDesc","cifvalue"]].copy()
+    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_period = gr.Textbox(label="دوره (YYYYMM، مثلا 202205)")
+    
     btn = gr.Button("نمایش کشورها")
-    out = gr.Dataframe(headers=["کشورهای واردکننده"], interactive=False)
+    out = gr.Dataframe(
+        headers=["reporterCode","reporterDesc","cifvalue"],
+        datatype=["number","text","number"],
+        interactive=False,
+    )
 
     btn.click(fn=get_importers,
               inputs=[inp_hs, inp_period],