Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import pandas as pd
 import comtradeapicall
 from huggingface_hub import InferenceClient
-from deep_translator import GoogleTranslator  # جایگزین googletrans
+from deep_translator import GoogleTranslator
 
 # کلید COMTRADE
 subscription_key = os.getenv("COMTRADE_API_KEY", "")
@@ -11,7 +11,7 @@ subscription_key = os.getenv("COMTRADE_API_KEY", "")
 hf_token = os.getenv("HF_API_TOKEN")
 
 client = InferenceClient(token=hf_token)
-translator = GoogleTranslator(source='en', target='fa')  # مترجم انگلیسی به فارسی
+translator = GoogleTranslator(source='en', target='fa')
 
 def get_importers(hs_code: str, year: str, month: str):
     period = f"{year}{int(month):02d}"
@@ -41,7 +41,7 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     prompt = (
         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 analysis in two complete paragraphs. The first paragraph should discuss the market opportunities and potential demand for this product in these countries. The second paragraph should offer strategic recommendations for exporters targeting these markets, focusing on trade strategies and risk management."
+        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)
@@ -50,7 +50,7 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         outputs = client.text_generation(
             prompt=prompt,
             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-            max_new_tokens=512
+            max_new_tokens=1024  # افزایش برای تکمیل جملات
         )
         print("خروجی مدل دریافت شد (به انگلیسی):")
         print(outputs)