Updated css + design

app.py

@@ -43,30 +43,59 @@ def model_chat(query: Optional[str], history: Optional[History]) -> Generator[Tu
         top_p=0.9,
         stream=True
     )
-    messages.append({'role': 'assistant', 'content': ''})
+    full_response = ''
     for completion in gen:
-        print(completion)
         text = completion.choices[0].delta.content
-        if not text: continue
-        messages[-1]['content'] += text
-        history = messages_to_history(messages)
-        yield '', history
+        full_response += text or ''
+        yield full_response
+        
+with gr.Blocks(css='''
+    .gr-group {direction: rtl;}
+    .chatbot{text-align:right;}
+  .dicta-header {
+    background-color: #f4f4f4;  /* Replace with desired background color */
+    border-radius: 10px;
+    padding: 20px;
+    text-align: center;
+    display: flex;
+    flex-direction: row;
+    align-items: center;
+  }
+
+  .dicta-logo {
+    width: 150px; /* Replace with actual logo width as desired */
+    height: auto;
+    margin-bottom: 20px;
+  }
+
+  .dicta-intro-text {
+    color: #333; /* Replace with desired text color */
+    margin-bottom: 20px;
+    text-align: center;
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    width: 100%;
+  }
+''') as demo:
+    gr.Markdown("""
+<div class="dicta-header">
+  <img src="file/dicta-logo.jpg" alt="Dicta Logo" class="dicta-logo" style="max-height: 75px">
+  <div class="dicta-intro-text">
+    <h1>DictaLM 2.0 - Instruct Chat Demo</h1>
+    <p>Welcome to the interactive demo of DictaLM-2.0. Explore the capabilities of our model and see how it can assist with your tasks.</p>
+    <p dir='rtl'> ברוכים הבאים לדמו האינטראקטיבי של DictaLM-2.0. חקרו את יכולות המודל שלנו וראו כיצד הוא יכול לסייע לכם במשימותיכם.</p>
+  </div>
+</div>
+""")
     
-with gr.Blocks() as demo:
-    gr.Markdown("""<center><font size=8>DictaLM2.0-Instruct Chat Demo</center>""")
+    interface = gr.ChatInterface(model_chat)
+    interface.chatbot.rtl = True
+    interface.textbox.placeholder = "הכנס שאלה בעברית (או באנגלית!)"
+    interface.textbox.rtl = True
+    interface.textbox.text_align = 'right'
+    interface.theme_css += '.gr-group {direction: rtl !important;}'
     
-    chatbot = gr.Chatbot(label='dicta-il/dictalm2.0-instruct', rtl=True)
-    textbox = gr.Textbox(placeholder="הכנס שאלה בעברית (או באנגלית!)", label='Input', rtl=True)
-
-    with gr.Row():
-        clear_history = gr.Button("🧹 Clear history")
-        sumbit = gr.Button("🚀 Send")
 
-    sumbit.click(model_chat,
-                 inputs=[textbox, chatbot],
-                 outputs=[textbox, chatbot])
-    clear_history.click(fn=clear_session,
-                        inputs=[],
-                        outputs=[textbox, chatbot])
 
-demo.queue(api_open=False).launch(max_threads=10,height=800, share=False)
+demo.queue(api_open=False).launch(max_threads=10,height=800, share=False, allowed_paths=['dicta-logo.jpg'], server_port=7861)