My initial chatbot

app.py

@@ -64,14 +64,12 @@ css = """
 }
 """
 
-# initialize the chat model and sets up the Gradio interface
+# gradio chatbot main function
 def registry(name, token, examples=None, **kwargs):
   client = genai.Client(api_key=token)
   chat_locks = {}  # Dictionary to hold locks for each user's chat
   chat_sessions = {} # Dictionary to hold each user chat
-  
 
-  # create a new chat session with the specified model and configuration
   def create_chat():
     return client.chats.create(
       model=name,
@@ -81,47 +79,53 @@ def registry(name, token, examples=None, **kwargs):
       ),
     )
 
-
   # send a user message to Gemini, streams the response back to the chatbot
   # and updates the history
   def stream_response(history, text, img, request: gr.Request):
-      user_id = request.client.host
-      if user_id not in chat_locks:
-        chat_locks[user_id] = Lock()
-        chat_sessions[user_id] = create_chat()
-
-      lock = chat_locks[user_id]
-      chat = chat_sessions[user_id]
-
-      try:
-        with lock:
-          if not img:
-            response_stream = chat.send_message_stream(text)
-          else:
-            try:
-              img = Image.open(img)
-              response_stream = chat.send_message_stream([text, img])
-            except Exception as e:
-              print(f"Error processing image: {str(e)}")
-              return
-
-          partial_message = ""
-          for chunk in response_stream:
-            if chunk.text:
-              partial_message += chunk.text
-              history[-1] = (history[-1][0], partial_message)
-              yield history
-              time.sleep(0.02)  # Small delay to ensure smooth streaming
-
-      except Exception as e:
-        print(f"Error in stream_response: {str(e)}")
-        return
+    user_id = request.client.host
+    if user_id not in chat_locks:
+      chat_locks[user_id] = Lock()
+      chat_sessions[user_id] = create_chat()
+
+    lock = chat_locks[user_id]
+    chat = chat_sessions[user_id]
+
+    try:
+      with lock:
+        if not img:
+          response_stream = chat.send_message_stream(
+            text
+          )
+        else:
+          try:
+            img = Image.open(img)
+            response_stream = chat.send_message_stream(
+              [text, img]
+            )
+          except Exception as e:
+            print(f"Error processing image: {str(e)}")
+            return
+
+        # Initialize response text
+        response_text = ""
+        
+        # Stream the response
+        for chunk in response_stream:
+          if chunk.text:
+            response_text += chunk.text
+            # Update the last message in history with the new content
+            history[-1] = (history[-1][0], response_text)
+            yield history
+
+    except Exception as e:
+      print(f"Error in stream_response: {str(e)}")
+      return
 
   print("Building the gradio app...")
   with gr.Blocks(css=css) as app:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
-    
+
     with gr.Row():
       image_box = gr.Image(type="filepath")
       chatbot = gr.Chatbot(
@@ -136,7 +140,7 @@ def registry(name, token, examples=None, **kwargs):
     )
 
     btn = gr.Button("Send")
-    
+
     # Update the event handlers to use streaming
     btn.click(
       fn=query_message,