Update app.py

app.py

@@ -54,16 +54,18 @@ def truncate_history(history):
     return history
     
 def get_full_context(input):
-  retrieved_documents = retriever.get_relevant_documents(input)
-  context = ""  # Initialize an empty string
+    retrieved_documents = retriever.get_relevant_documents(input)
+    context = ""  
 
-  file_path = "./docs/Troubleshooting_Table.csv"
-  data = pd.read_csv(file_path)
-  for doc in retrieved_documents:
-      index = doc.metadata['index']
-      context += str(data.iloc[[index]])
+    file_path = "./docs/Troubleshooting_Table.csv"
+    data = pd.read_csv(file_path)
 
-  return context
+    for doc in retrieved_documents:
+        index = doc.metadata['index']
+        row_string = data.iloc[[index]].to_string(index=False)
+        context += row_string + "\n\n" 
+
+    return context
 
 
 is_first_run = True  # Flag to check if it's the first run
@@ -74,7 +76,6 @@ def predict(history, input):
     global is_first_run  # Use the global flag
     if is_first_run:
         context = get_full_context(input)
-        print(type(context))
         print(context)  # For debugging
         is_first_run = False  # Set the flag to False after the first run
     else: