Update app.py

app.py

@@ -33,17 +33,17 @@ def stream_chat_with_rag(
     print(f"History: {history}")
 
     # Build the conversation prompt including system prompt and history
-    conversation = system_prompt + "\n\n" + "For Client: " + client_name + "\n"
+    conversation = system_prompt + "\n\n" + f"For Client: {client_name}\n"
     for user_input, assistant_response in history:
         conversation += f"User: {user_input}\nAssistant: {assistant_response}\n"
     conversation += f"User: {message}\nAssistant:"
 
     # Prepare the data to send to the API
+    # Remove 'history' from the payload since the API does not accept it
     api_payload = {
-        "message": message,
-        "history": history,
+        "message": conversation,  # Include the history in the message
         "client_name": client_name,
-        "system_prompt": system_prompt,
+        "system_prompt": "",  # Optionally set to empty if included in message
         "num_retrieved_docs": num_retrieved_docs,
         "num_docs_final": num_docs_final,
         "temperature": temperature,
@@ -59,12 +59,11 @@ def stream_chat_with_rag(
         **api_payload
     )
 
-    # Assuming the API returns the assistant's reply
-    # If the API returns a tuple (answer, relevant_docs), extract the answer
+    # Extract the assistant's reply
     if isinstance(response, tuple):
         answer = response[0]
     else:
-        answer = response  # If it's already the assistant's reply
+        answer = response
 
     # Debugging statements
     print("The Answer in stream_chat_with_rag:")
@@ -76,6 +75,7 @@ def stream_chat_with_rag(
     # Return the updated history
     return history
 
+
 # Function to handle PDF processing API call
 def process_pdf(pdf_file):
     return client.predict(