Update app.py

app.py

@@ -29,39 +29,33 @@ print("MODEL READY")
 
 
 
-@spaces.GPU
-def generate(message, history):
+def format_answer(text):
 
-    prompt = """
-You are Rezaeian StatsAI.
-You are a professional statistics assistant.
+    # مرتب سازی خروجی فارسی
+    text = text.replace("\n", "<br>")
 
-Rules:
-- Answer in Persian when user writes Persian.
-- Write statistical formulas clearly.
-- Keep numbers and English letters separated correctly.
-- Use readable mathematical notation.
-"""
+    # فرمول های LaTeX
+    text = text.replace("\\[", "<div class='formula'>")
+    text = text.replace("\\]", "</div>")
+
+    text = text.replace("$$", "<div class='formula'>")
+
+    return text
 
 
-    for user, assistant in history:
-        prompt += f"""
 
-User:
-{user}
+@spaces.GPU
+def generate(message, history):
 
-Assistant:
-{assistant}
-"""
+    prompt = ""
 
 
-    prompt += f"""
+    for user, assistant in history:
 
-User:
-{message}
+        prompt += f"User: {user}\nAssistant: {assistant}\n"
 
-Assistant:
-"""
+
+    prompt += f"User: {message}\nAssistant:"
 
 
     inputs = tokenizer(
@@ -71,8 +65,8 @@ Assistant:
 
 
     inputs = {
-        k:v.to(model.device)
-        for k,v in inputs.items()
+        k: v.to(model.device)
+        for k, v in inputs.items()
     }
 
 
@@ -95,14 +89,15 @@ Assistant:
 
     answer = result.split("Assistant:")[-1]
 
-    return answer
+
+    return format_answer(answer)
 
 
 
 demo = gr.ChatInterface(
     fn=generate,
     title="Rezaeian StatsAI",
-    description="دستیار هوشمند آمار و احتمال مهندسی"
+    description="دستیار هوشمند آمار و فرمول‌های آماری"
 )
 
 
@@ -116,18 +111,33 @@ demo.launch(
         max-width: 1200px !important;
     }
 
-    textarea {
+
+    .message {
         direction: rtl !important;
         text-align: right !important;
+        font-family: Tahoma, Arial, sans-serif !important;
         font-size:18px !important;
+        line-height:2;
     }
 
-    .message {
+
+    textarea {
         direction: rtl !important;
         text-align:right !important;
         font-size:18px !important;
     }
 
+
+    .formula {
+        direction:ltr !important;
+        text-align:center !important;
+        font-family: Cambria Math, Times New Roman, serif !important;
+        font-size:20px !important;
+        padding:12px;
+        margin:12px;
+    }
+
+
     footer {
         display:none !important;
     }