Update app.py

app.py

@@ -15,21 +15,26 @@ def respond(
     temperature,
     top_p,
 ):
-    messages = [{"role": "system", "content": system_message}]
+    messages = [{"role": "system", "content": system_message or "You are a friendly Chatbot."}]
 
-    # Add history to messages
+    # Add history to messages, ensuring no None values
     for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+        user_message = val[0] if val[0] is not None else ""
+        assistant_message = val[1] if val[1] is not None else ""
 
-    # Add the current user message
-    messages.append({"role": "user", "content": message})
+        if user_message:
+            messages.append({"role": "user", "content": user_message})
+        if assistant_message:
+            messages.append({"role": "assistant", "content": assistant_message})
+
+    # Add the current user message, ensure it's not None
+    if message:
+        messages.append({"role": "user", "content": message})
 
     # Generate the response using the Mistral model
     response = llm.create_chat_completion(messages=messages)
-
+    print("response:", response)
+    
     return response["choices"][0]["message"]["content"]  # Adjust based on your model's output format
 
 # Set up Gradio Chat Interface