Update app.py

app.py

@@ -1,11 +1,18 @@
+import os
 import gradio as gr
 import pandas as pd
 import comtradeapicall
-import os
-from datetime import datetime
+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", "")
-proxy_url = None
+hf_token         = os.getenv("HF_API_TOKEN", None)
+proxy_url        = None
+
+inference = InferenceApi(repo_id="google/gemma-2b", token=hf_token)
 
 def get_importers(hs_code: str, year: str, month: str):
     period = f"{year}{int(month):02d}"
@@ -28,7 +35,21 @@ def get_importers(hs_code: str, year: str, month: str):
     result.columns = ["کد کشور","نام کشور","ارزش CIF"]
     return result
 
-current_year = datetime.now().year
+def provide_advice(table, hs_code: str, year: str, month: str):
+    if not table:
+        return "ابتدا باید اطلاعات واردات را نمایش دهید."
+    df = pd.DataFrame(table, columns=["کد کشور","نام کشور","ارزش CIF"])
+    table_str = df.to_markdown(index=False)
+    period = f"{year}/{int(month):02d}"
+    prompt = (
+        f"جدول زیر اطلاعات کشورهایی است که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
+        f"{table_str}\n\n"
+        "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
+    )
+    response = inference(prompt)
+    return response
+
+current_year = pd.Timestamp.now().year
 years = [str(y) for y in range(2000, current_year+1)]
 months = [str(m) for m in range(1, 13)]
 
@@ -37,14 +58,27 @@ with gr.Blocks() as demo:
     with gr.Row():
         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(datetime.now().month))
-    btn = gr.Button("نمایش اطلاعات")
-    out = gr.Dataframe(
+        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"],
-        interactive=False
+        interactive=False,
+    )
+    btn_show.click(
+        fn=get_importers,
+        inputs=[inp_hs, inp_year, inp_month],
+        outputs=out_table
+    )
+
+    btn_advice = gr.Button("ارائه مشاوره تخصصی")
+    out_advice = gr.Textbox(label="مشاوره تخصصی")
+
+    btn_advice.click(
+        fn=provide_advice,
+        inputs=[out_table, inp_hs, inp_year, inp_month],
+        outputs=out_advice
     )
-    btn.click(get_importers, inputs=[inp_hs, inp_year, inp_month], outputs=out)
 
 if __name__ == "__main__":
     demo.launch()