Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import spaces
 import torch
+import re
 
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
@@ -29,17 +30,47 @@ print("MODEL READY")
 
 
 
+def clean_text(text):
+
+    # مرتب کردن فاصله های فرمول
+    text = text.replace(" ,", ",")
+    text = text.replace(" .", ".")
+
+    # نمایش بهتر فرمول ها
+    text = text.replace("$", "")
+
+    return text
+
+
+
 @spaces.GPU
 def generate(message, history):
 
-    prompt = ""
+    prompt = """
+شما Rezaeian StatsAI هستید.
+دستیار تخصصی آمار هستید.
+فرمول ها را دقیق و خوانا بنویسید.
+برای فارسی راست چین پاسخ دهید.
+"""
 
     for user, assistant in history:
-        prompt += f"User: {user}\n"
-        prompt += f"Assistant: {assistant}\n"
 
+        prompt += f"""
+کاربر:
+{user}
 
-    prompt += f"User: {message}\nAssistant:"
+پاسخ:
+{assistant}
+"""
+
+
+    prompt += f"""
+
+کاربر:
+{message}
+
+پاسخ:
+"""
 
 
     inputs = tokenizer(
@@ -49,8 +80,8 @@ def generate(message, history):
 
 
     inputs = {
-        k: v.to(model.device)
-        for k, v in inputs.items()
+        k:v.to(model.device)
+        for k,v in inputs.items()
     }
 
 
@@ -59,109 +90,120 @@ def generate(message, history):
         output = model.generate(
             **inputs,
             max_new_tokens=512,
-            temperature=0.7,
+            temperature=0.6,
             top_p=0.9,
-            do_sample=True,
             repetition_penalty=1.1
         )
 
 
-    text = tokenizer.decode(
+    result = tokenizer.decode(
         output[0],
         skip_special_tokens=True
     )
 
 
-    answer = text.split("Assistant:")[-1]
-
+    answer = result.split("پاسخ:")[-1]
 
-    return answer.strip()
+    return clean_text(answer)
 
 
 
 css = """
 
-.gradio-container {
-
-    max-width: 900px !important;
-    margin: auto !important;
+body{
+direction:rtl;
+}
 
+.gradio-container{
+max-width:1200px !important;
 }
 
 
-textarea {
+.chatbot{
+font-size:18px !important;
+height:700px !important;
+}
 
-    direction: rtl !important;
-    text-align: right !important;
 
+.message{
+direction:rtl !important;
+text-align:right !important;
+font-family:Tahoma, Arial !important;
 }
 
 
-.message {
+textarea{
+direction:rtl !important;
+font-size:18px !important;
+}
 
-    direction: rtl !important;
-    text-align: right !important;
 
+footer{
+display:none !important;
 }
 
-
 """
 
 
 with gr.Blocks() as demo:
 
 
-    gr.Markdown(
-        """
-        # 🤖 Rezaeian StatsAI
-
-        دستیار هوش مصنوعی آمار و تحلیل داده
-        """
-    )
+    gr.HTML("""
+    <h1 style="
+    text-align:center;
+    direction:rtl;">
+    🤖 Rezaeian StatsAI
+    </h1>
+    
+    <p style="
+    text-align:center;
+    direction:rtl;
+    font-size:18px;">
+    دستیار هوشمند آمار و فرمول های آماری
+    </p>
+    """)
 
 
     chatbot = gr.Chatbot(
-        height=450
+        height=700
     )
 
 
-    with gr.Row():
+    textbox = gr.Textbox(
+        placeholder="سوال آماری خود را بنویسید...",
+        lines=2
+    )
 
-        textbox = gr.Textbox(
-            placeholder="سوال خود را بنویسید...",
-            scale=8
-        )
 
+    with gr.Row():
 
         send = gr.Button(
             "ارسال"
         )
 
-
         clear = gr.Button(
             "پاک کردن"
         )
 
 
-
     send.click(
         generate,
         inputs=[textbox, chatbot],
-        outputs=[textbox, chatbot]
+        outputs=chatbot
     )
 
 
     textbox.submit(
         generate,
         inputs=[textbox, chatbot],
-        outputs=[textbox, chatbot]
+        outputs=chatbot
     )
 
 
     clear.click(
-        lambda: [],
-        inputs=None,
-        outputs=chatbot
+        lambda: None,
+        None,
+        chatbot
     )