Update app.py

app.py

@@ -16,9 +16,6 @@ client = Client("on1onmangoes/CNIHUB101324v10", hf_token=HF_TOKEN)
 # 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,
     history: list,
@@ -46,12 +43,60 @@ def stream_chat_with_rag(
         penalty=penalty,
         api_name="/chat"
     )
-    
+
+    # Debugging statements
+    print(f"API Response Type: {type(response)}")
+    print(f"API Response: {response}")
+
+    # Ensure response is a string
+    if isinstance(response, list):
+        # If response is a list, extract the assistant's reply
+        response = response[0] if response else ""
+
     # Update the conversation history
     history = history + [(message, response)]
+
+    # Return the updated history
+    return history
+
+
+
+
+# # Function to handle chat API call
+# # 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"
+#     )
+    
+#     # Update the conversation history
+#     history = history + [(message, response)]
     
-    # Return the assistant's reply and the updated history
-    return "", history
+#     # Return the assistant's reply and the updated history
+#     return "", history
 
 
 # # Function to handle chat API call