Update app.py

app.py

@@ -5,11 +5,16 @@ import os
 import json
 import gradio as gr 
 
-def construct_index(file):
+def load_index(file_path):
+    # Load the index from the saved file
+    index = GPTSimpleVectorIndex.load_from_disk(file_path)
+    return index
+
+def construct_index(file_path):
     PDFReader = download_loader("PDFReader")
 
     loader = PDFReader()
-    documents = loader.load_data(file=Path(file.name))
+    documents = loader.load_data(file=Path(file_path))
 
     # Construct a simple vector index
     index = GPTSimpleVectorIndex.from_documents(documents)
@@ -19,10 +24,10 @@ def construct_index(file):
 
     return index
 
-def qabot(file, input_text):
-    # Load the index from the uploaded file
-    index = construct_index(file)
+# Initialize the index variable with the index from the saved file
+index = load_index('index.json')
 
+def qabot(input_text):
     # Query the index with the user's input text
     response = index.query(input_text, response_mode="compact")
     return response.response
@@ -31,10 +36,11 @@ def qabot(file, input_text):
 file_upload = gr.inputs.File(label="Upload PDF file")
 
 # Change the input components of the function to the file upload component and a text box for user input
-iface = gr.Interface(fn=qabot, inputs=[file_upload, gr.inputs.Textbox(lines=7, label='Enter your query')], outputs="text", title="Custom-trained QA Application")
+iface = gr.Interface(fn=qabot, inputs=gr.inputs.Textbox(lines=7, label='Enter your query'), outputs="text", title="Custom-trained QA Application")
 
 # Add a separate interface to update the index
 def update_index(file):
+    global index
     index = construct_index(file)
     return "Index generated from uploaded file: {}".format(file.name)
 
@@ -42,4 +48,4 @@ update_index_interface = gr.Interface(update_index, inputs=file_upload, outputs=
 
 # Launch both interfaces
 iface.launch()
-update_index_interface.launch()
+update_index_interface.launch()