Update main.py

main.py

@@ -108,7 +108,6 @@ def simulate_text_conversation(caller_number: str, initial_message: str, convers
         system_prompt = (
             f"You are a call assistant. The unknown caller's text message is '{initial_message}'.\n"
             "Simulate a multi-turn conversation. Do not mention that you are a virtual assistant. Act like a human"
-            
         )
     else:
         system_prompt = "You are a call assistant. Do not mention that you are a virtual assistant. Act like a human"
@@ -118,15 +117,13 @@ def simulate_text_conversation(caller_number: str, initial_message: str, convers
     for entry in chat_history:
         if entry.get("role") in {"system", "user", "assistant"}:
             messages.append({"role": entry["role"], "content": entry["content"]})
-    messages.append({"role": "user", "content": initial_message})
-    
+    # The initial user message is already saved in chat history
     assistant_response = llm.invoke(messages)
     if hasattr(assistant_response, "content"):
         assistant_response = assistant_response.content
     else:
         assistant_response = str(assistant_response)
     
-    update_chat_history(caller_number, "user", initial_message)
     update_chat_history(caller_number, "assistant", assistant_response)
     
     return assistant_response
@@ -150,15 +147,13 @@ def simulate_call_conversation(caller_number: str, initial_message: str, convers
     for entry in chat_history:
         if entry.get("role") in {"system", "user", "assistant"}:
             messages.append({"role": entry["role"], "content": entry["content"]})
-    messages.append({"role": "user", "content": initial_message})
-    
+    # The transcribed message is already in chat history
     assistant_response = llm.invoke(messages)
     if hasattr(assistant_response, "content"):
         assistant_response = assistant_response.content
     else:
         assistant_response = str(assistant_response)
     
-    update_chat_history(caller_number, "user", initial_message)
     update_chat_history(caller_number, "assistant", assistant_response)
     
     return assistant_response
@@ -192,6 +187,9 @@ def get_contact(phone: str):
 
 @app.post("/incoming-message")
 def process_incoming_message(incoming: IncomingMessage):
+    # Save the incoming message in MongoDB chat history
+    update_chat_history(incoming.phone, "user", incoming.message)
+    
     if contacts_collection.find_one({"phone": incoming.phone}):
         return {
             "status": "primary",