Update app.py

app.py

@@ -16,35 +16,38 @@ def respond(
     top_p,
     image=None
 ):
-    try:
-        messages = [{"role": "system", "content": system_message}]
+    messages = [{"role": "system", "content": system_message}]
 
-        for val in history:
-            if val[0]:
-                messages.append({"role": "user", "content": val[0]})
-            if val[1]:
-                messages.append({"role": "assistant", "content": val[1]})
+    for val in history:
+        if val[0]:
+            messages.append({"role": "user", "content": val[0]})
+        if val[1]:
+            messages.append({"role": "assistant", "content": val[1]})
 
-        messages.append({"role": "user", "content": message})
+    messages.append({"role": "user", "content": message})
 
-        if image:
-            # Handle single image input
-            image_b64 = image_to_base64(image)  # Convert image to base64
-            messages.append({"role": "user", "content": "Image uploaded", "image": image_b64})
+    if image:
+        # Convert image to base64
+        image_b64 = image_to_base64(image)
+        messages.append({"role": "user", "content": "Image uploaded", "image": image_b64})
 
-        # Call Hugging Face model for response
-        response = ""
-        for message in client.chat_completion(
-            messages,
-            max_tokens=max_tokens,
-            stream=True,
-            temperature=temperature,
-            top_p=top_p,
-        ):
-            token = message.choices[0].delta.content
+    # Call Hugging Face model for response
+    responses = []
+    for message in client.chat_completion(
+        messages,
+        max_tokens=max_tokens,
+        stream=True,
+        temperature=temperature,
+        top_p=top_p,
+    ):
+        token = message.choices[0].delta.content
+
+        responses.append(token)
+
+    # Yield each response sequentially
+    for response in responses:
+        yield response
 
-            response += token
-            yield response
 
     except Exception as e:
         print(f"Error in respond function: {str(e)}")