Update app.py

app.py

@@ -28,24 +28,15 @@ model.eval()
 print("MODEL READY")
 
 
+
 @spaces.GPU
 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]}\n"
-            prompt += f"Assistant: {item[1]}\n"
+    for user, assistant in history:
+        prompt += f"User: {user}\n"
+        prompt += f"Assistant: {assistant}\n"
 
 
     prompt += f"User: {message}\nAssistant:"
@@ -75,13 +66,13 @@ def generate(message, history):
         )
 
 
-    result = tokenizer.decode(
+    text = tokenizer.decode(
         output[0],
         skip_special_tokens=True
     )
 
 
-    answer = result.split("Assistant:")[-1]
+    answer = text.split("Assistant:")[-1]
 
 
     return answer.strip()
@@ -90,11 +81,6 @@ def generate(message, history):
 
 css = """
 
-body {
-    direction: rtl;
-}
-
-
 .gradio-container {
 
     max-width: 900px !important;
@@ -122,7 +108,7 @@ textarea {
 """
 
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
 
 
     gr.Markdown(
@@ -135,70 +121,38 @@ with gr.Blocks(css=css) as demo:
 
 
     chatbot = gr.Chatbot(
-        height=450,
-        type="messages"
+        height=450
     )
 
 
     with gr.Row():
 
-
         textbox = gr.Textbox(
             placeholder="سوال خود را بنویسید...",
-            scale=8,
-            container=False
+            scale=8
         )
 
 
         send = gr.Button(
-            "ارسال",
-            scale=1
+            "ارسال"
         )
 
 
         clear = gr.Button(
-            "پاک کردن",
-            scale=1
+            "پاک کردن"
         )
 
 
 
-    def chat(message, history):
-
-        answer = generate(
-            message,
-            history
-        )
-
-
-        history = history + [
-
-            {
-                "role": "user",
-                "content": message
-            },
-
-            {
-                "role": "assistant",
-                "content": answer
-            }
-
-        ]
-
-
-        return "", history
-
-
-
     send.click(
-        chat,
+        generate,
         inputs=[textbox, chatbot],
         outputs=[textbox, chatbot]
     )
 
 
     textbox.submit(
-        chat,
+        generate,
         inputs=[textbox, chatbot],
         outputs=[textbox, chatbot]
     )
@@ -213,5 +167,6 @@ with gr.Blocks(css=css) as demo:
 
 
 demo.launch(
-    server_name="0.0.0.0"
+    server_name="0.0.0.0",
+    css=css
 )