Update app.py

app.py

@@ -5,14 +5,17 @@ import comtradeapicall
 from huggingface_hub import InferenceClient
 from deep_translator import GoogleTranslator
 
+
 # کلید COMTRADE
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
 # توکن Hugging Face
 hf_token = os.getenv("HF_API_TOKEN")
 
+
 client = InferenceClient(token=hf_token)
 translator = GoogleTranslator(source='en', target='fa')
 
+
 def get_importers(hs_code: str, year: str, month: str):
     period = f"{year}{int(month):02d}"
     df = comtradeapicall.previewFinalData(
@@ -23,7 +26,7 @@ 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)
@@ -31,34 +34,32 @@ def get_importers(hs_code: str, year: str, month: str):
           .sort_values("cifvalue", ascending=False)
     )
     result.columns = ["کد کشور", "نام کشور", "ارزش CIF"]
-    
-    product_name = df['cmdDesc'].iloc[0] if 'cmdDesc' in df.columns else "برنج"
-    return result, product_name
+    return result
 
-def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str, product_name: str):
+
+def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
     if table_data is None or table_data.empty:
         return "ابتدا باید اطلاعات واردات را نمایش دهید."
     table_str = table_data.to_string(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = (
-        f"The following table shows countries that imported the product '{product_name}' with HS code {hs_code} during the period {period}:\n"
+        f"The following table shows countries that imported a product with HS code {hs_code} during the period {period}:\n"
         f"{table_str}\n\n"
         f"Please provide a detailed and comprehensive analysis in two paragraphs. The first paragraph should discuss market opportunities, potential demand, and specific cultural or economic factors influencing the demand for this product in these countries. The second paragraph should offer actionable strategic recommendations for exporters, including detailed trade strategies, risk management techniques, and steps to establish local partnerships."
     )
     print("پرامپت ساخته‌شده:")
     print(prompt)
     try:
-        print("در حال فراخوانی مدل google/gemma-2b-it با متد conversational...")
-        # استفاده از متد conversational
-        conversation = client.conversational(
-            text=prompt,
-            model="google/gemma-2b-it",
-            max_new_tokens=1024
+        print("در حال فراخوانی مدل mistralai/Mixtral-8x7B-Instruct-v0.1...")
+        outputs = client.text_generation(
+            prompt=prompt,
+            model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+            max_new_tokens=1024  # افزایش برای تکمیل جملات
         )
-        outputs = conversation['generated_text'] if 'generated_text' in conversation else conversation
         print("خروجی مدل دریافت شد (به انگلیسی):")
         print(outputs)
 
+
         # ترجمه خروجی به فارسی
         translated_outputs = translator.translate(outputs)
         print("خروجی ترجمه‌شده به فارسی:")
@@ -69,14 +70,16 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         print(error_msg)
         return error_msg
 
+
 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)]
 
+
 with gr.Blocks() as demo:
     gr.Markdown("## نمایش کشورهایی که یک کالا را وارد کرده‌اند")
     with gr.Row():
-        inp_hs = gr.Textbox(label="HS Code", value="1006")
+        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("نمایش اطلاعات")
@@ -85,20 +88,19 @@ with gr.Blocks() as demo:
         datatype=["number", "text", "number"],
         interactive=True,
     )
-    btn_show.click(
-        fn=get_importers,
-        inputs=[inp_hs, inp_year, inp_month],
-        outputs=[out_table, gr.State()]
-    )
+    btn_show.click(get_importers, [inp_hs, inp_year, inp_month], out_table)
+
 
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
     out_advice = gr.Textbox(label="مشاوره تخصصی", lines=6)
 
+
     btn_advice.click(
-        fn=provide_advice,
-        inputs=[out_table, inp_hs, inp_year, inp_month, gr.State()],
+        provide_advice,
+        inputs=[out_table, inp_hs, inp_year, inp_month],
         outputs=out_advice
     )
 
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()