Update app.py

app.py

@@ -10,6 +10,21 @@ sessions = {}
 PRIMARY_SYSTEM_INSTRUCTIONS = "You are P-MSQ (Messaging Service Query), a friendly AI Chatbot that can help in any situations"
 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 render_avatars(userid):
+    try:
+        response = requests.post(
+            'https://host.palple.polrambora.com/userexistence',
+            json={
+                'userid': userid
+            }
+        )
+
+        if response.status_code === 200:
+            response_json = response.json()
+            return response_json["avatar"]["link"]
+    except Exception as e:
+        return None
+
 
 def authorize(user, api_key, system_message):
     test_data = {
@@ -25,6 +40,7 @@ def authorize(user, api_key, system_message):
 
         if response.status_code == 200:
             response_json = response.json()
+            avatar = render_avatars(user) or USER_PIC_PATH
             
             if api_key not in sessions:
                 sessions[api_key] = {
@@ -33,6 +49,7 @@ def authorize(user, api_key, system_message):
                         "authorization": api_key,
                         "Content-Type": 'application/json'
                     },
+                    "avatar": avatar,
                     "system_message": system_message 
                 }
             return True
@@ -85,6 +102,7 @@ def respond(message, api_key, max_tokens, top_p, temperature):
     else:
         return history, "Error: " + response.json().get("error", "Unknown error occurred.")
 
+
 def render_message(history):
     messages_html = """
     <div id="chatbox-container" class="chatbox" style="height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex; flex-direction: column-reverse;">
@@ -171,7 +189,7 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow: auto; border: 1px solid #
         last_message_state = gr.State("")  
 
         def user_interaction(message, history, api_key, max_tokens, top_p, temperature):
-            loading_message = history + [(message, "Loading...", "You", "P-ALPLE", USER_PIC_PATH, ASSISTANT_PIC_PATH)]
+            loading_message = history + [(message, "Loading...", "You", "P-ALPLE", sessions[api_key]["avatar"], ASSISTANT_PIC_PATH)]
             yield render_message(loading_message), loading_message, ""
 
             history, assistant_reply = respond(message, api_key, max_tokens, top_p, temperature)