Update app.py

app.py

@@ -45,7 +45,7 @@ def format_chat_history(messages: list) -> list:
             "role": "user",
             "parts": [message.get("content", "")]
             })
-        elif message.get("role") == "model" :
+        elif message.get("role") == "assistant" :
             formatted_history.append({
             "role": "model",
             "parts": [message.get("content", "")]
@@ -95,7 +95,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
         
         messages.append(
             ChatMessage(
-                role="model",
+                role="assistant",
                 content=response_buffer
             )
         )
@@ -113,7 +113,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
                 print(f"\n=== Complete Thought ===\n{thought_buffer}")
 
                 messages[-1] = ChatMessage(
-                    role="model",
+                    role="assistant",
                     content=thought_buffer,
                     metadata={"title": "⚙️ Thinking: *The thoughts produced by the model are experimental"}
                 )
@@ -125,7 +125,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
 
                 messages.append(
                     ChatMessage(
-                        role="model",
+                        role="assistant",
                         content=response_buffer
                     )
                 )
@@ -137,7 +137,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
                 print(f"\n=== Response Chunk ===\n{current_chunk}")
 
                 messages[-1] = ChatMessage(
-                    role="model",
+                    role="assistant",
                     content=response_buffer
                 )
 
@@ -147,7 +147,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
                 print(f"\n=== Thinking Chunk ===\n{current_chunk}")
 
                 messages[-1] = ChatMessage(
-                    role="model",
+                    role="assistant",
                     content=thought_buffer,
                     metadata={"title": "⚙️ Thinking: *The thoughts produced by the model are experimental"}
                 )