diginoron commited on
Commit
3caec61
·
verified ·
1 Parent(s): 9a96732

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -31,11 +31,11 @@ def get_importers(hs_code: str, year: str, month: str):
31
  result.columns = ["کد کشور","نام کشور","ارزش CIF"]
32
  return result
33
 
34
- def provide_advice(table_data, hs_code: str, year: str, month: str):
35
- if not table_data:
 
36
  return "ابتدا باید اطلاعات واردات را نمایش دهید."
37
- df = pd.DataFrame(table_data, columns=["کد کشور","نام کشور","ارزش CIF"])
38
- table_str = df.to_string(index=False)
39
  period = f"{year}/{int(month):02d}"
40
  prompt = (
41
  f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
@@ -59,7 +59,7 @@ with gr.Blocks() as demo:
59
  out_table = gr.Dataframe(
60
  headers=["کد کشور","نام کشور","ارزش CIF"],
61
  datatype=["number","text","number"],
62
- interactive=True, # حتما interactive=True تا داده به صورت لیست برگردد
63
  )
64
  btn_show.click(get_importers, inputs=[inp_hs, inp_year, inp_month], outputs=out_table)
65
 
 
31
  result.columns = ["کد کشور","نام کشور","ارزش CIF"]
32
  return result
33
 
34
+ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
35
+ # بررسی خالی‌بودن DataFrame
36
+ if table_data is None or table_data.empty:
37
  return "ابتدا باید اطلاعات واردات را نمایش دهید."
38
+ table_str = table_data.to_string(index=False)
 
39
  period = f"{year}/{int(month):02d}"
40
  prompt = (
41
  f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
 
59
  out_table = gr.Dataframe(
60
  headers=["کد کشور","نام کشور","ارزش CIF"],
61
  datatype=["number","text","number"],
62
+ interactive=True,
63
  )
64
  btn_show.click(get_importers, inputs=[inp_hs, inp_year, inp_month], outputs=out_table)
65