Update app.py

app.py

@@ -3,9 +3,8 @@ import pandas as pd
 import comtradeapicall
 import os
 
-# کلید API (برای preview نیازی نیست)
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
-proxy_url        = None
+proxy_url = None
 
 def get_importers(hs_code: str, period: str):
     df = comtradeapicall.previewFinalData(
@@ -18,32 +17,27 @@ def get_importers(hs_code: str, period: str):
     )
     if df is None or df.empty:
         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)
+    result = (
+        df.groupby(["reporterCode","reporterDesc"], as_index=False)
+          .agg({"cifvalue":"sum"})
+          .sort_values("cifvalue", ascending=False)
+    )
+    result.columns = ["کد کشور","نام کشور","ارزش CIF"]
     return result
 
 with gr.Blocks() as demo:
-    gr.Markdown("## جست‌وجوی واردکننده‌های یک HS-Code")
+    gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
     with gr.Row():
-        inp_hs     = gr.Textbox(label="HS Code (مثلاً 010121)")
-        inp_period = gr.Textbox(label="دوره (YYYYMM، مثلاً 202205)")
-    btn = gr.Button("نمایش کشورها")
+        inp_hs = gr.Textbox(label="HS Code")
+        inp_period = gr.Textbox(label="دوره (YYYYMM)")
+    btn = gr.Button("نمایش")
     out = gr.Dataframe(
-        headers=["کد کشور", "نام کشور", "ارزش CIF"],
+        headers=["کد کشور","نام کشور","ارزش CIF"],
         datatype=["number","text","number"],
-        interactive=False,
+        interactive=False
     )
-
-    btn.click(fn=get_importers,
-              inputs=[inp_hs, inp_period],
-              outputs=out)
+    btn.click(get_importers, inputs=[inp_hs, inp_period], outputs=out)
 
 if __name__ == "__main__":
     demo.launch()