app.py

import gradio as gr
import pandas as pd
import comtradeapicall
import os

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',
        partnerCode=None, partner2Code=None,
        customsCode=None, motCode=None,
        maxRecords=500, includeDesc=True,
        proxy_url=proxy_url
    )
    if df is None or df.empty:
        # اگر خالی بود، یک DataFrame خالی برمی‌گردانیم
        return pd.DataFrame(columns=["reporterCode","reporterDesc","cifvalue"])
    
    # ۲. فیلتر و مرتب‌سازی
    df = df[df['cifvalue'] > 0]
    df = df.sort_values('cifvalue', ascending=False)
    
    # ۳. ستون‌های مورد نظر
    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)")
    
    btn = gr.Button("نمایش کشورها")
    out = gr.Dataframe(
        headers=["reporterCode","reporterDesc","cifvalue"],
        datatype=["number","text","number"],
        interactive=False,
    )

    btn.click(fn=get_importers,
              inputs=[inp_hs, inp_period],
              outputs=out)

if __name__ == "__main__":
    demo.launch()