Update app.py

app.py

@@ -109,22 +109,14 @@ def convert_messages_to_tuples(history: List[Dict[str, str]]) -> List[Tuple[str,
     This robust version correctly handles histories that start with an assistant message.
     """
     tuple_history = []
-    # Find the start of the actual conversation (the first user message)
-    start_index = 0
+    # Iterate through the history to find user messages
     for i, msg in enumerate(history):
         if msg['role'] == 'user':
-            start_index = i
-            break
-    
-    # Group the rest of the messages into (user, assistant) pairs
-    user_msg_content = None
-    for i in range(start_index, len(history)):
-        if history[i]['role'] == 'user':
-            user_msg_content = history[i]['content']
-        elif history[i]['role'] == 'assistant' and user_msg_content is not None:
-            tuple_history.append((user_msg_content, history[i]['content']))
-            user_msg_content = None # Reset to find the next pair
-            
+            # Once a user message is found, check if the next message is from the assistant
+            if i + 1 < len(history) and history[i+1]['role'] == 'assistant':
+                user_content = msg['content']
+                assistant_content = history[i+1]['content']
+                tuple_history.append((user_content, assistant_content))
     return tuple_history
 
 # --- Gradio UI ---