Update app.py

app.py

@@ -31,11 +31,10 @@ def get_importers(hs_code: str, year: str, month: str):
     result.columns = ["کد کشور","نام کشور","ارزش CIF"]
     return result
 
-def provide_advice(table, hs_code: str, year: str, month: str):
-    if not table or len(table) == 0:
+def provide_advice(table: pd.DataFrame, hs_code: str, year: str, month: str):
+    if table is None or table.empty:
         return "ابتدا باید اطلاعات واردات را نمایش دهید."
-    df = pd.DataFrame(table, columns=["کد کشور","نام کشور","ارزش CIF"])
-    table_md = df.to_markdown(index=False)
+    table_md = table.to_markdown(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = (
         f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
@@ -61,14 +60,16 @@ with gr.Blocks() as demo:
         datatype=["number","text","number"],
         interactive=False,
     )
-    btn_show.click(get_importers, [inp_hs, inp_year, inp_month], out_table)
+    btn_show.click(get_importers, inputs=[inp_hs, inp_year, inp_month], outputs=out_table)
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
     out_advice = gr.Textbox(label="مشاوره تخصصی", lines=5)
 
-    btn_advice.click(provide_advice,
-                     inputs=[out_table, inp_hs, inp_year, inp_month],
-                     outputs=out_advice)
+    btn_advice.click(
+        provide_advice,
+        inputs=[out_table, inp_hs, inp_year, inp_month],
+        outputs=out_advice
+    )
 
 if __name__ == "__main__":
     demo.launch()