diginoron commited on
Commit
9a96732
·
verified ·
1 Parent(s): 27d5142

Update app.py

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