Update app.py

app.py

@@ -28,31 +28,19 @@ model.eval()
 print("MODEL READY")
 
 
-
-def format_answer(text):
-
-    # تبدیل چند فرمول رایج آماری به LaTeX
-    text = text.replace("β0", "$\\beta_0$")
-    text = text.replace("β1", "$\\beta_1$")
-    text = text.replace("R2", "$R^2$")
-    text = text.replace("σ2", "$\\sigma^2$")
-    text = text.replace("x̄", "$\\bar{x}$")
-
-    return text
-
-
-
 @spaces.GPU
 def generate(message, history):
 
     prompt = ""
 
-    for user, assistant in history:
-
-        prompt += (
-            f"User: {user}\n"
-            f"Assistant: {assistant}\n"
-        )
+    for item in history:
+        if isinstance(item, dict):
+            if item["role"] == "user":
+                prompt += f"User: {item['content']}\n"
+            else:
+                prompt += f"Assistant: {item['content']}\n"
+        else:
+            prompt += f"User: {item[0]}\nAssistant: {item[1]}\n"
 
 
     prompt += f"User: {message}\nAssistant:"
@@ -65,24 +53,20 @@ 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()
     }
 
 
     with torch.no_grad():
 
         output = model.generate(
-
             **inputs,
-
             max_new_tokens=512,
-
             temperature=0.7,
-
             top_p=0.9,
-
-            do_sample=True
+            do_sample=True,
+            repetition_penalty=1.1
         )
 
 
@@ -94,62 +78,91 @@ def generate(message, history):
 
     answer = result.split("Assistant:")[-1]
 
-
-    return format_answer(answer)
+    return answer.strip()
 
 
 
 css = """
-
-footer {
-display:none !important;
+body {
+    direction: rtl;
 }
 
-
 .gradio-container {
+    direction: rtl;
+    text-align: right;
+}
 
-direction:rtl !important;
+textarea {
+    direction: rtl !important;
+    text-align: right !important;
+}
 
+.message {
+    direction: rtl;
+    text-align: right;
 }
 
+"""
 
-.message {
 
-direction:rtl !important;
+with gr.Blocks(css=css) as demo:
 
-text-align:right !important;
+    gr.Markdown(
+        """
+        # 🤖 Rezaeian StatsAI
+        ### دستیار هوش مصنوعی آمار و تحلیل داده
+        """
+    )
 
-unicode-bidi:plaintext;
 
-}
+    chatbot = gr.Chatbot(
+        height=600
+    )
 
 
-textarea {
+    msg = gr.Textbox(
+        placeholder="سوال خود را بنویسید...",
+        lines=2
+    )
 
-direction:rtl !important;
 
-text-align:right !important;
+    clear = gr.Button("پاک کردن")
 
-}
 
+    def respond(message, history):
 
-code, pre {
+        answer = generate(message, history)
 
-direction:ltr !important;
+        history.append(
+            {
+                "role":"user",
+                "content":message
+            }
+        )
 
-text-align:left !important;
+        history.append(
+            {
+                "role":"assistant",
+                "content":answer
+            }
+        )
 
-}
+        return "", history
 
-"""
 
 
-demo = gr.ChatInterface(
-    fn=generate,
-    title="Rezaeian StatsAI",
-    description="دستیار هوش مصنوعی آمار",
-    css=css
-)
+    msg.submit(
+        respond,
+        [msg, chatbot],
+        [msg, chatbot]
+    )
+
+
+    clear.click(
+        lambda: [],
+        None,
+        chatbot
+    )