Update app.py

app.py

@@ -22,14 +22,18 @@ def construct_index(directory_path):
 
     return index
 
+'''
 def ask_ai():
     index = GPTSimpleVectorIndex.load_from_disk('index.json')
     while True: 
         query = input("What do you want to ask? ")
         response = index.query(query, response_mode="compact")
         display(Markdown(f"Response: <b>{response.response}</b>"))
-
-
+'''
+def ask_ai(question):
+    index = GPTSimpleVectorIndex.load_from_disk('index.json')
+    response = index.query(question, response_mode="compact")
+    return response.response
 
 os.environ["OPENAI_API_KEY"] = "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"