Update message type parsing

yankeexe · Dec 18, 2024 · 6e73abf · 6e73abf
1 parent d876cff
commit 6e73abf
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ollama_manager/ui/image_chat.py b/ollama_manager/ui/image_chat.py
@@ -96,11 +96,11 @@ def run():
     st.title("🤖 Ollama Manager: Vision Chat")
 
     for message in st.session_state["messages"]:
-        match message["role"]:
-            case "assistant":
-                st.chat_message("assistant").write(message["content"])
-            case "user":
-                st.chat_message("human").write(message["content"])
+        message_role = message["role"]
+        if message_role == "assistant":
+            st.chat_message("assistant").write(message["content"])
+        if message_role == "user":
+            st.chat_message("human").write(message["content"])
 
     chat_input = st.chat_input(
         placeholder="Write your message...",