Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import comtradeapicall
 from huggingface_hub import InferenceClient
 
-# کلید COMTRADE (فعلاً برای preview نیازی به آن نداریم)
+# کلید COMTRADE
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 # توکن Hugging Face
 hf_token = os.getenv("HF_API_TOKEN")
@@ -21,14 +21,14 @@ def get_importers(hs_code: str, year: str, month: str):
         maxRecords=500, includeDesc=True
     )
     if df is None or df.empty:
-        return pd.DataFrame(columns=["کد کشور","نام کشور","ارزش CIF"])
+        return pd.DataFrame(columns=["کد کشور", "نام کشور", "ارزش CIF"])
     df = df[df['cifvalue'] > 0]
     result = (
-        df.groupby(["reporterCode","reporterDesc"], as_index=False)
-          .agg({"cifvalue":"sum"})
+        df.groupby(["reporterCode", "reporterDesc"], as_index=False)
+          .agg({"cifvalue": "sum"})
           .sort_values("cifvalue", ascending=False)
     )
-    result.columns = ["کد کشور","نام کشور","ارزش CIF"]
+    result.columns = ["کد کشور", "نام کشور", "ارزش CIF"]
     return result
 
 def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
@@ -41,14 +41,13 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         f"{table_str}\n\n"
         "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
     )
-    # فراخوانی text_generation
+    # فراخوانی text_generation با پارامتر درست
     outputs = client.text_generation(
         model="google/gemma-2b",
-        inputs=prompt,
-        parameters={"max_new_tokens":256}
+        prompt=prompt,
+        max_new_tokens=256
     )
-    # خروجی اولین شات
-    return outputs[0]["generated_text"]
+    return outputs
 
 current_year = pd.Timestamp.now().year
 years = [str(y) for y in range(2000, current_year+1)]
@@ -57,13 +56,13 @@ months = [str(m) for m in range(1, 13)]
 with gr.Blocks() as demo:
     gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
     with gr.Row():
-        inp_hs    = gr.Textbox(label="HS Code")
-        inp_year  = gr.Dropdown(choices=years, label="سال", value=str(current_year))
+        inp_hs = gr.Textbox(label="HS Code")
+        inp_year = gr.Dropdown(choices=years, label="سال", value=str(current_year))
         inp_month = gr.Dropdown(choices=months, label="ماه", value=str(pd.Timestamp.now().month))
     btn_show = gr.Button("نمایش اطلاعات")
     out_table = gr.Dataframe(
-        headers=["کد کشور","نام کشور","ارزش CIF"],
-        datatype=["number","text","number"],
+        headers=["کد کشور", "نام کشور", "ارزش CIF"],
+        datatype=["number", "text", "number"],
         interactive=True,
     )
     btn_show.click(get_importers, [inp_hs, inp_year, inp_month], out_table)
@@ -78,4 +77,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()