Update app.py

app.py

@@ -5,27 +5,28 @@ import gradio as gr
 client = Client("dicta-il/dictalm2.0-instruct-demo")
 
 def chat_with_model(history):
-    prompt = history[-1][0]
+    prompt = history[-1]["content"]
     result = client.predict(
         message=prompt,
         api_name="/chat"
     )
-    return history + [(prompt, result)]
+    return history + [{"role": "user", "content": prompt}, {"role": "bot", "content": result}]
 
-# יצירת ממשק מתקדם עם Gradio בצורת צ'ט-בוט
+# יצירת ממשק מתקדם עם Gradio בצורת צ'ט-בוט עם לוגו של DICTA
 with gr.Blocks(theme="default") as demo:
+    gr.Image(value="dicta-logo.jpg", elem_id="logo", width=200, height=100)
     gr.Markdown("# צ'אט עם מודל DictaLM", elem_id="title")
-    chatbot = gr.Chatbot(label="צ'אט עם מודל DictaLM")
+    chatbot = gr.Chatbot(label="צ'אט עם מודל DictaLM", type="messages")
     with gr.Row():
         user_input = gr.Textbox(placeholder="הכנס את ההודעה שלך כאן...", label="", lines=1)
         send_button = gr.Button("שלח")
     
     def user_chat(history, message):
-        return history + [(message, "")], ""
+        return history + [{"role": "user", "content": message}], ""
 
     user_input.submit(user_chat, inputs=[chatbot, user_input], outputs=[chatbot, user_input], queue=False)
     user_input.submit(chat_with_model, inputs=chatbot, outputs=chatbot)
     send_button.click(user_chat, inputs=[chatbot, user_input], outputs=[chatbot, user_input], queue=False)
     send_button.click(chat_with_model, inputs=chatbot, outputs=chatbot)
 
-demo.launch()
+demo.launch()