Update app.py

app.py

@@ -64,14 +64,18 @@ def respond(
     # Format system context with the selected prompt strategy
     formatted_system_message = system_context.format(prompt_strategy=prompt_template.template)
     
-    # Prepare messages
+    # Prepare messages with proper format
     messages = [{"role": "system", "content": formatted_system_message}]
+    
+    # Add conversation history
     for user_msg, assistant_msg in history:
         if user_msg:
-            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "user", "content": str(user_msg)})
         if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
-    messages.append({"role": "user", "content": message})
+            messages.append({"role": "assistant", "content": str(assistant_msg)})
+    
+    # Add current message
+    messages.append({"role": "user", "content": str(message)})
 
     # Get generation parameters
     generation_params = prompt_template.parameters if not override_params else {
@@ -80,14 +84,16 @@ def respond(
         "top_p": top_p
     }
 
-    # Generate response using the assistant
-    for response in assistant.generate_response(
-        prompt_template=prompt_template,
-        generation_params=generation_params,
-        stream=True,
-        messages=messages
-    ):
-        yield response
+    try:
+        for response in assistant.generate_response(
+            prompt_template=prompt_template,
+            generation_params=generation_params,
+            stream=True,
+            messages=messages
+        ):
+            yield response
+    except Exception as e:
+        yield f"Error: {str(e)}"
 
 with gr.Blocks() as demo:
     with gr.Row():