Update app.py

app.py

@@ -26,39 +26,40 @@ 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
+    if image:
+        # Convert image to base64
+        if isinstance(image, Image.Image):
+            image_b64 = image_to_base64(image)
             messages.append({"role": "user", "content": "Image uploaded", "image": image_b64})
+        else:
+            # Handle multiple images if necessary
+            for img in image:
+                image_b64 = image_to_base64(img)
+                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
+    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
 
-            response += token
-            yield response
-
-    except Exception as e:
-        print(f"Error in respond function: {str(e)}")
-        yield f"Error occurred: {str(e)}"
+        response += token
+        yield response
 
 
 # Debugging print statements