Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import spaces
 import torch
-import re
 
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
@@ -30,46 +29,38 @@ 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 = """
-شما Rezaeian StatsAI هستید.
-دستیار تخصصی آمار هستید.
-فرمول ها را دقیق و خوانا بنویسید.
-برای فارسی راست چین پاسخ دهید.
+You are Rezaeian StatsAI.
+You are a professional statistics assistant.
+
+Rules:
+- Answer in Persian when user writes Persian.
+- Write statistical formulas clearly.
+- Keep numbers and English letters separated correctly.
+- Use readable mathematical notation.
 """
 
-    for user, assistant in history:
 
+    for user, assistant in history:
         prompt += f"""
-کاربر:
+
+User:
 {user}
 
-پاسخ:
+Assistant:
 {assistant}
 """
 
 
     prompt += f"""
 
-کاربر:
+User:
 {message}
 
-پاسخ:
+Assistant:
 """
 
 
@@ -90,9 +81,9 @@ def generate(message, history):
         output = model.generate(
             **inputs,
             max_new_tokens=512,
-            temperature=0.6,
+            temperature=0.7,
             top_p=0.9,
-            repetition_penalty=1.1
+            do_sample=True
         )
 
 
@@ -102,113 +93,44 @@ def generate(message, history):
     )
 
 
-    answer = result.split("پاسخ:")[-1]
-
-    return clean_text(answer)
-
-
-
-css = """
-
-body{
-direction:rtl;
-}
-
-.gradio-container{
-max-width:1200px !important;
-}
-
-
-.chatbot{
-font-size:18px !important;
-height:700px !important;
-}
-
-
-.message{
-direction:rtl !important;
-text-align:right !important;
-font-family:Tahoma, Arial !important;
-}
-
-
-textarea{
-direction:rtl !important;
-font-size:18px !important;
-}
-
-
-footer{
-display:none !important;
-}
-
-"""
-
-
-with gr.Blocks() as demo:
-
-
-    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=700
-    )
-
-
-    textbox = gr.Textbox(
-        placeholder="سوال آماری خود را بنویسید...",
-        lines=2
-    )
-
+    answer = result.split("Assistant:")[-1]
 
-    with gr.Row():
+    return answer
 
-        send = gr.Button(
-            "ارسال"
-        )
 
-        clear = gr.Button(
-            "پاک کردن"
-        )
 
+demo = gr.ChatInterface(
+    fn=generate,
+    title="Rezaeian StatsAI",
+    description="دستیار هوشمند آمار و احتمال مهندسی"
+)
 
-    send.click(
-        generate,
-        inputs=[textbox, chatbot],
-        outputs=chatbot
-    )
 
 
-    textbox.submit(
-        generate,
-        inputs=[textbox, chatbot],
-        outputs=chatbot
-    )
+demo.launch(
+    server_name="0.0.0.0",
+    css="""
 
+    .gradio-container {
+        direction: rtl;
+        max-width: 1200px !important;
+    }
 
-    clear.click(
-        lambda: None,
-        None,
-        chatbot
-    )
+    textarea {
+        direction: rtl !important;
+        text-align: right !important;
+        font-size:18px !important;
+    }
 
+    .message {
+        direction: rtl !important;
+        text-align:right !important;
+        font-size:18px !important;
+    }
 
+    footer {
+        display:none !important;
+    }
 
-demo.launch(
-    server_name="0.0.0.0",
-    css=css
+    """
 )