Update app.py

app.py

@@ -11,7 +11,8 @@ headers = {
     "Content-Type": "application/json",
 }
 
-ASSISTANT_PIC_PATH = "./API.png"  
+ASSISTANT_PIC_PATH = "https://huggingface.co/spaces/PLRMB/P-MSQ-API-PREVIEW/resolve/main/API.png"  
+USER_PIC_PATH = "https://huggingface.co/spaces/PLRMB/P-MSQ-API-PREVIEW/resolve/main/usr.png"
 
 def respond(message, history, system_message, max_tokens, top_p, temperature):
     messages = []
@@ -47,7 +48,7 @@ def respond(message, history, system_message, max_tokens, top_p, temperature):
     if response.status_code == 200:
         response_json = response.json()
         assistant_reply = response_json["msq"]["message"][0]
-        history.append((message, assistant_reply, "User", "Assistant", None, ASSISTANT_PIC_PATH))
+        history.append((message, assistant_reply, "You", "P-ALPLE", USER_PIC_PATH, ASSISTANT_PIC_PATH))
         return history, assistant_reply
     else:
         return history, "Error: " + response.json().get("error", "Unknown error occurred.")
@@ -56,13 +57,13 @@ def render_message(history):
     messages_html = ""
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
         if user_message:
-            messages_html += f"<div style='display: flex; align-items: center;'>"
+            messages_html += f"<div style='display: flex; align-items: center; margin-bottom: 10px;'>"
             if user_pic:
                 messages_html += f"<img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>"
             messages_html += f"<b>{user_profile}:</b> {user_message}</div><br>"
 
         if assistant_message:
-            messages_html += f"<div style='display: flex; align-items: center;'>"
+            messages_html += f"<div style='display: flex; align-items: center; margin-bottom: 10px;'>"
             if assistant_pic:
                 messages_html += f"<img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>"
             messages_html += f"<b>{assistant_profile}:</b> {assistant_message}</div><br>"