Update app.py

app.py

@@ -2,12 +2,13 @@ import gradio as gr
 import pandas as pd
 import comtradeapicall
 import os
+from datetime import datetime
 
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 proxy_url = None
 
-def get_importers(hs_code: str, date):
-    period = f"{date.year:04d}{date.month:02d}"
+def get_importers(hs_code: str, year: str, month: str):
+    period = f"{year}{int(month):02d}"
     df = comtradeapicall.previewFinalData(
         typeCode='C', freqCode='M', clCode='HS', period=period,
         reporterCode=None, cmdCode=hs_code, flowCode='M',
@@ -27,18 +28,23 @@ def get_importers(hs_code: str, date):
     result.columns = ["کد کشور","نام کشور","ارزش CIF"]
     return result
 
+current_year = datetime.now().year
+years = [str(y) for y in range(2000, current_year+1)]
+months = [str(m) for m in range(1, 13)]
+
 with gr.Blocks() as demo:
     gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
     with gr.Row():
-        inp_hs = gr.Textbox(label="HS Code")
-        inp_date = gr.DatePicker(label="تاریخ (انتخاب روزی در ماه/سال)")
+        inp_hs    = gr.Textbox(label="HS Code")
+        inp_year  = gr.Dropdown(choices=years, label="سال", value=str(current_year))
+        inp_month = gr.Dropdown(choices=months, label="ماه", value=str(datetime.now().month))
     btn = gr.Button("نمایش اطلاعات")
     out = gr.Dataframe(
         headers=["کد کشور","نام کشور","ارزش CIF"],
         datatype=["number","text","number"],
         interactive=False
     )
-    btn.click(get_importers, [inp_hs, inp_date], out)
+    btn.click(get_importers, inputs=[inp_hs, inp_year, inp_month], outputs=out)
 
 if __name__ == "__main__":
     demo.launch()