Update app.py

app.py

@@ -6,11 +6,17 @@ import sys
 import os
 
 
+cache = {}
+cache.put("RetroFeedback", GPTSimpleVectorIndex.load_from_disk('RetroFeedback.json'))
+cache.put("Snowflake", GPTSimpleVectorIndex.load_from_disk('Snowflake.json'))
+cache.put("Datadog", GPTSimpleVectorIndex.load_from_disk('Datadog.json'))
+cache.put("Databricks", GPTSimpleVectorIndex.load_from_disk('Databricks.json'))
+
 def chatbot(indexName, input_text):
     """
     Chatbot function that takes in a prompt and returns a response
     """
-    index = GPTSimpleVectorIndex.load_from_disk(indexName+'.json')
+    index = cache.get(indexName)
     response = index.query(input_text, response_mode="compact")
     return response.response