Update app.py

app.py

@@ -20,6 +20,22 @@ def compare_responses(message, model1, approach1, model2, approach2, system_mess
     response2 = respond(message, [], model2, approach2, system_message, max_tokens, temperature, top_p)
     return response1, response2
 
+def parse_conversation(messages):
+    system_prompt = ""
+    conversation = []
+    
+    for message in messages:
+        role = message['role']
+        content = message['content']
+        
+        if role == 'system':
+            system_prompt = content
+        elif role in ['user', 'assistant']:
+            conversation.append(f"{role.capitalize()}: {content}")
+    
+    initial_query = "\n".join(conversation)
+    return system_prompt, initial_query
+
 def respond(message, history, model, approach, system_message, max_tokens, temperature, top_p):
     client = OpenAI(api_key=API_KEY, base_url="https://openrouter.ai/api/v1")
     messages = [{"role": "system", "content": system_message}]
@@ -42,8 +58,7 @@ def respond(message, history, model, approach, system_message, max_tokens, tempe
         )
         return response.choices[0].message.content
     else:
-        system_prompt = system_message
-        initial_query = message
+        system_prompt, initial_query = parse_conversation(messages)
         if approach == 'rto':
             final_response, _ = round_trip_optimization(system_prompt, initial_query, client, model)
         elif approach == 'z3':