Update app.py

app.py

@@ -34,15 +34,15 @@ def get_importers(hs_code: str, year: str, month: str):
 def provide_advice(table: pd.DataFrame, hs_code: str, year: str, month: str):
     if table is None or table.empty:
         return "ابتدا باید اطلاعات واردات را نمایش دهید."
-    table_md = table.to_markdown(index=False)
+    table_md = table.to_string(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = (
         f"جدول زیر کشورهایی را نشان می‌دهد که کالا با کد HS {hs_code} را در دوره {period} وارد کرده‌اند:\n"
         f"{table_md}\n\n"
         "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
     )
-    response = inference(prompt)
-    return response
+    raw = inference(prompt, raw_response=True)
+    return raw.text
 
 current_year = pd.Timestamp.now().year
 years = [str(y) for y in range(2000, current_year+1)]