Update app.py

app.py

@@ -6,7 +6,8 @@ import os
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 proxy_url = None
 
-def get_importers(hs_code: str, period: str):
+def get_importers(hs_code: str, date):
+    period = f"{date.year:04d}{date.month:02d}"
     df = comtradeapicall.previewFinalData(
         typeCode='C', freqCode='M', clCode='HS', period=period,
         reporterCode=None, cmdCode=hs_code, flowCode='M',
@@ -30,14 +31,14 @@ with gr.Blocks() as demo:
     gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
     with gr.Row():
         inp_hs = gr.Textbox(label="HS Code")
-        inp_period = gr.Textbox(label="دوره (YYYYMM)")
-    btn = gr.Button("نمایش")
+        inp_date = gr.DatePicker(label="تاریخ (انتخاب ماه و سال)", type="date")
+    btn = gr.Button("نمایش اطلاعات")
     out = gr.Dataframe(
         headers=["کد کشور","نام کشور","ارزش CIF"],
         datatype=["number","text","number"],
         interactive=False
     )
-    btn.click(get_importers, inputs=[inp_hs, inp_period], outputs=out)
+    btn.click(get_importers, inputs=[inp_hs, inp_date], outputs=out)
 
 if __name__ == "__main__":
     demo.launch()