Update app.py

app.py

@@ -37,9 +37,9 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     table_str = table_data.to_string(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = (
-        f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {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 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."
     )
     print("پرامپت ساخته‌شده:")
     print(prompt)
@@ -47,8 +47,8 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         print("در حال فراخوانی مدل mistralai/Mixtral-8x7B-Instruct-v0.1...")
         outputs = client.text_generation(
             prompt=prompt,
-            model="mistralai/Mixtral-8x7B-Instruct-v0.1",  # مدل جایگزین
-            max_new_tokens=256
+            model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+            max_new_tokens=512  # افزایش طول خروجی
         )
         print("خروجی مدل دریافت شد:")
         print(outputs)