Update app.py

app.py

@@ -2,13 +2,14 @@ import os
 import gradio as gr
 import pandas as pd
 import comtradeapicall
-from huggingface_hub import InferenceApi
+from huggingface_hub import InferenceClient
 
+# کلید COMTRADE (فعلاً برای preview نیازی به آن نداریم)
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
-hf_token         = os.getenv("HF_API_TOKEN", None)
-proxy_url        = None
+# توکن Hugging Face
+hf_token = os.getenv("HF_API_TOKEN")
 
-inference = InferenceApi(repo_id="google/gemma-2b", token=hf_token)
+client = InferenceClient(token=hf_token)
 
 def get_importers(hs_code: str, year: str, month: str):
     period = f"{year}{int(month):02d}"
@@ -17,8 +18,7 @@ def get_importers(hs_code: str, year: str, month: str):
         reporterCode=None, cmdCode=hs_code, flowCode='M',
         partnerCode=None, partner2Code=None,
         customsCode=None, motCode=None,
-        maxRecords=500, includeDesc=True,
-        proxy_url=proxy_url
+        maxRecords=500, includeDesc=True
     )
     if df is None or df.empty:
         return pd.DataFrame(columns=["کد کشور","نام کشور","ارزش CIF"])
@@ -32,7 +32,6 @@ def get_importers(hs_code: str, year: str, month: str):
     return result
 
 def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
-    # بررسی خالی‌بودن DataFrame
     if table_data is None or table_data.empty:
         return "ابتدا باید اطلاعات واردات را نمایش دهید."
     table_str = table_data.to_string(index=False)
@@ -42,8 +41,14 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         f"{table_str}\n\n"
         "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
     )
-    raw = inference(prompt, raw_response=True)
-    return raw.text
+    # فراخوانی text_generation
+    outputs = client.text_generation(
+        model="google/gemma-2b",
+        inputs=prompt,
+        parameters={"max_new_tokens":256}
+    )
+    # خروجی اولین شات
+    return outputs[0]["generated_text"]
 
 current_year = pd.Timestamp.now().year
 years = [str(y) for y in range(2000, current_year+1)]
@@ -61,10 +66,10 @@ with gr.Blocks() as demo:
         datatype=["number","text","number"],
         interactive=True,
     )
-    btn_show.click(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="مشاوره تخصصی", lines=5)
+    out_advice = gr.Textbox(label="مشاوره تخصصی", lines=6)
 
     btn_advice.click(
         provide_advice,