diginoron commited on
Commit
6df290f
·
verified ·
1 Parent(s): ab94ac0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -2,12 +2,15 @@ import gradio as gr
2
  import pandas as pd
3
  import comtradeapicall
4
  import os
 
 
5
 
6
  subscription_key = os.getenv("COMTRADE_API_KEY", "")
7
  proxy_url = None
8
 
9
- def get_importers(hs_code: str, date):
10
- period = f"{date.year:04d}{date.month:02d}"
 
11
  df = comtradeapicall.previewFinalData(
12
  typeCode='C', freqCode='M', clCode='HS', period=period,
13
  reporterCode=None, cmdCode=hs_code, flowCode='M',
@@ -30,8 +33,8 @@ def get_importers(hs_code: str, date):
30
  with gr.Blocks() as demo:
31
  gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
32
  with gr.Row():
33
- inp_hs = gr.Textbox(label="HS Code")
34
- inp_date = gr.DatePicker(label="تاریخ (انتخاب ماه و سال)", type="date")
35
  btn = gr.Button("نمایش اطلاعات")
36
  out = gr.Dataframe(
37
  headers=["کد کشور","نام کشور","ارزش CIF"],
 
2
  import pandas as pd
3
  import comtradeapicall
4
  import os
5
+ from gradio.components import DatePicker
6
+ from datetime import datetime
7
 
8
  subscription_key = os.getenv("COMTRADE_API_KEY", "")
9
  proxy_url = None
10
 
11
+ def get_importers(hs_code: str, date_str):
12
+ # date_str comes in "YYYY-MM-DD" format; extract YYYYMM
13
+ period = date_str[:7].replace("-", "")
14
  df = comtradeapicall.previewFinalData(
15
  typeCode='C', freqCode='M', clCode='HS', period=period,
16
  reporterCode=None, cmdCode=hs_code, flowCode='M',
 
33
  with gr.Blocks() as demo:
34
  gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
35
  with gr.Row():
36
+ inp_hs = gr.Textbox(label="HS Code")
37
+ inp_date = DatePicker(label="تاریخ (انتخاب یک روز در ماه و سال مورد نظر)")
38
  btn = gr.Button("نمایش اطلاعات")
39
  out = gr.Dataframe(
40
  headers=["کد کشور","نام کشور","ارزش CIF"],