Update app.py

app.py

@@ -4,10 +4,6 @@ import pandas as pd
 import comtradeapicall
 from huggingface_hub import InferenceApi
 
-# Environment variables:
-# COMTRADE_API_KEY  -> برای COMTRADE (preview نیازی نیست اما برای آینده)
-# HF_API_TOKEN     -> برای Hugging Face Inference API
-
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 hf_token         = os.getenv("HF_API_TOKEN", None)
 proxy_url        = None
@@ -36,14 +32,14 @@ def get_importers(hs_code: str, year: str, month: str):
     return result
 
 def provide_advice(table, hs_code: str, year: str, month: str):
-    if not table:
+    if not table or len(table) == 0:
         return "ابتدا باید اطلاعات واردات را نمایش دهید."
     df = pd.DataFrame(table, columns=["کد کشور","نام کشور","ارزش CIF"])
-    table_str = df.to_markdown(index=False)
+    table_md = df.to_markdown(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = (
-        f"جدول زیر اطلاعات کشورهایی است که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
-        f"{table_str}\n\n"
+        f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
+        f"{table_md}\n\n"
         "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
     )
     response = inference(prompt)
@@ -65,20 +61,14 @@ with gr.Blocks() as demo:
         datatype=["number","text","number"],
         interactive=False,
     )
-    btn_show.click(
-        fn=get_importers,
-        inputs=[inp_hs, inp_year, inp_month],
-        outputs=out_table
-    )
+    btn_show.click(get_importers, [inp_hs, inp_year, inp_month], out_table)
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
-    out_advice = gr.Textbox(label="مشاوره تخصصی")
+    out_advice = gr.Textbox(label="مشاوره تخصصی", lines=5)
 
-    btn_advice.click(
-        fn=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()