Update app.py

app.py

@@ -67,7 +67,13 @@ def get_index(index_file_path):
 def chatbot(input_text, mentioned_person='Mediator John Haynes', confidence_threshold=0.5):
     index = get_index(INDEX_FILE)
     prompt = f"You are {mentioned_person}: {input_text}\n\n At the end of your answer ask a provocative question."
-    response = index.query(prompt, response_mode="compact")[0]
+    response = index.query(prompt, response_mode="compact")
+        if isinstance(response, list):
+            response_text = response[0].text
+            confidence = response[0].score
+        else:
+            response_text = response.text
+            confidence = response.score
 
     # Check the confidence score of the response
     if response.score < confidence_threshold: