Update app.py

app.py

@@ -1,6 +1,7 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import datetime
+from pathlib import Path
 
 # Initialize the InferenceClient
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -26,7 +27,7 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
     )
     now = datetime.datetime.now()
     formatted_time = now.strftime("%H:%M:%S, %B %d, %Y")
-    system_prompt = f"System time: {formatted_time}"
+    system_prompt = f"System time: {formatted_time}. System time: {formatted_time}. Instructions: Everything else is from the user. You are Milo, an AI assistant created by ConvoLite in 2024 (he/him). Be friendly and empathetic, matching the user's tone. Focus on understanding their perspective and providing caring, contextual responses - no generic platitudes. Keep it conversational, not overly formal."
     formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
@@ -42,9 +43,11 @@ additional_inputs = [
     gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Penalize repeated tokens")
 ]
 
+avatar_images = ("https://i.postimg.cc/pXjKKVXG/user-circle.png", "https://i.postimg.cc/qq04Yz93/CL3.png")
+
 gr.ChatInterface(
-    fn=generate,
-    chatbot=gr.Chatbot(show_label=True, show_share_button=False, show_copy_button=True, likeable=True, layout="panel", height="auto"),
+    fn=chat,
+    chatbot=gr.Chatbot(show_label=True, show_share_button=False, show_copy_button=True, likeable=True, layout="panel", height="auto", avatar_images=avatar_images),
     additional_inputs=additional_inputs,
     title="ConvoLite",
     submit_btn="➢",