Update app.py

app.py

@@ -34,13 +34,18 @@ def generate(message, history):
     prompt = ""
 
     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: {item[0]}\n"
+            prompt += f"Assistant: {item[1]}\n"
 
 
     prompt += f"User: {message}\nAssistant:"
@@ -78,90 +83,131 @@ def generate(message, history):
 
     answer = result.split("Assistant:")[-1]
 
+
     return answer.strip()
 
 
 
 css = """
+
 body {
     direction: rtl;
 }
 
+
 .gradio-container {
-    direction: rtl;
-    text-align: right;
+
+    max-width: 900px !important;
+    margin: auto !important;
+
 }
 
+
 textarea {
+
     direction: rtl !important;
     text-align: right !important;
+
 }
 
+
 .message {
-    direction: rtl;
-    text-align: right;
+
+    direction: rtl !important;
+    text-align: right !important;
+
 }
 
+
 """
 
 
 with gr.Blocks(css=css) as demo:
 
+
     gr.Markdown(
         """
         # 🤖 Rezaeian StatsAI
-        ### دستیار هوش مصنوعی آمار و تحلیل داده
+
+        دستیار هوش مصنوعی آمار و تحلیل داده
         """
     )
 
 
     chatbot = gr.Chatbot(
-        height=600
+        height=450,
+        type="messages"
     )
 
 
-    msg = gr.Textbox(
-        placeholder="سوال خود را بنویسید...",
-        lines=2
-    )
+    with gr.Row():
 
 
-    clear = gr.Button("پاک کردن")
+        textbox = gr.Textbox(
+            placeholder="سوال خود را بنویسید...",
+            scale=8,
+            container=False
+        )
 
 
-    def respond(message, history):
+        send = gr.Button(
+            "ارسال",
+            scale=1
+        )
 
-        answer = generate(message, history)
 
-        history.append(
-            {
-                "role":"user",
-                "content":message
-            }
+        clear = gr.Button(
+            "پاک کردن",
+            scale=1
+        )
+
+
+
+    def chat(message, history):
+
+        answer = generate(
+            message,
+            history
         )
 
-        history.append(
+
+        history = history + [
+
             {
-                "role":"assistant",
-                "content":answer
+                "role": "user",
+                "content": message
+            },
+
+            {
+                "role": "assistant",
+                "content": answer
             }
-        )
+
+        ]
+
 
         return "", history
 
 
 
-    msg.submit(
-        respond,
-        [msg, chatbot],
-        [msg, chatbot]
+    send.click(
+        chat,
+        inputs=[textbox, chatbot],
+        outputs=[textbox, chatbot]
+    )
+
+
+    textbox.submit(
+        chat,
+        inputs=[textbox, chatbot],
+        outputs=[textbox, chatbot]
     )
 
 
     clear.click(
         lambda: [],
-        None,
-        chatbot
+        inputs=None,
+        outputs=chatbot
     )