Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from datetime import datetime
 import random
+import uuid
 
 # Shared state to store messages
 messages = []
 
-# Generate a random color for each user
+# Dictionary to store user colors
 user_colors = {}
 
 def get_user_color(user_id):
@@ -13,19 +14,20 @@ def get_user_color(user_id):
         user_colors[user_id] = f"#{random.randint(0, 0xFFFFFF):06x}"
     return user_colors[user_id]
 
-def chat(message, history):
+def chat(message, history, user_id):
     global messages
     
-    # Generate a unique user ID (in a real app, this would be managed by authentication)
-    user_id = f"User_{len(user_colors) + 1}"
+    if not user_id:
+        # Generate a new user ID if one doesn't exist
+        user_id = str(uuid.uuid4())
     
     # Add the new message to the shared state
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     user_color = get_user_color(user_id)
-    messages.append([None, f"<span style='color: {user_color};'>{user_id}</span>: {message}<br><small>{timestamp}</small>"])
+    messages.append([None, f"<span style='color: {user_color};'>User_{user_id[:4]}</span>: {message}<br><small>{timestamp}</small>"])
     
-    # Return the updated chat history
-    return "", messages
+    # Return the updated chat history and user_id
+    return "", messages, user_id
 
 def get_updates(history):
     global messages
@@ -42,12 +44,13 @@ with gr.Blocks(css="#chatbot {height: 400px; overflow-y: auto;}") as demo:
     chatbot = gr.Chatbot(elem_id="chatbot")
     msg = gr.Textbox(label="Type your message here")
     clear = gr.Button("Clear")
+    user_id = gr.State(value='')  # Add a state component to store the user ID
 
-    msg.submit(chat, [msg, chatbot], [msg, chatbot])
+    msg.submit(chat, [msg, chatbot, user_id], [msg, chatbot, user_id])
     clear.click(lambda: [], outputs=[chatbot])
     
     # Add an update function that runs every second
-    demo.load(get_updates, inputs=chatbot, outputs=chatbot, every=1)
+    demo.load(get_updates, inputs=chatbot, outputs=chatbot, every=0.05)
 
 # Launch the app
 if __name__ == "__main__":