Update app.py

app.py

@@ -10,6 +10,14 @@ HF_TOKEN = os.getenv("HF_TOKEN")  # Replace with your actual token if not using
 client = Client("on1onmangoes/CNIHUB101324v10", hf_token=HF_TOKEN)
 # on1onmangoes/CNIHUB101324v10
 
+
+# Here's how you can fix it:
+
+# Update the conversation history within the function.
+# Return the updated history along with any other required outputs.
+
+
+# Function to handle chat API call
 # Function to handle chat API call
 def stream_chat_with_rag(
     message: str,
@@ -38,8 +46,44 @@ def stream_chat_with_rag(
         penalty=penalty,
         api_name="/chat"
     )
-    # Return the assistant's reply
-    return response
+    
+    # Update the conversation history
+    history = history + [(message, response)]
+    
+    # Return the assistant's reply and the updated history
+    return "", history
+
+
+# # Function to handle chat API call
+# def stream_chat_with_rag(
+#     message: str,
+#     history: list,
+#     client_name: str,
+#     system_prompt: str,
+#     num_retrieved_docs: int,
+#     num_docs_final: int,
+#     temperature: float,
+#     max_new_tokens: int,
+#     top_p: float,
+#     top_k: int,
+#     penalty: float,
+# ):
+#     # Use the parameters provided by the UI
+#     response = client.predict(
+#         message=message,
+#         client_name=client_name,
+#         system_prompt=system_prompt,
+#         num_retrieved_docs=num_retrieved_docs,
+#         num_docs_final=num_docs_final,
+#         temperature=temperature,
+#         max_new_tokens=max_new_tokens,
+#         top_p=top_p,
+#         top_k=top_k,
+#         penalty=penalty,
+#         api_name="/chat"
+#     )
+#     # Return the assistant's reply
+#     return response
 
 # Function to handle PDF processing API call
 def process_pdf(pdf_file):