Edited

app.py

@@ -10,6 +10,71 @@ from llama_index import (
 from llama_index.node_parser import SimpleNodeParser
 from langchain import OpenAI
 import gradio as gr
+import sys
+import os
+import os.path
+import shutil
+
+sys.path.append("/my_functions")
+
+os.environ["OPENAI_API_KEY"] = "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"
+
+# Defining the parameters for the index
+max_input_size = 4096
+num_outputs = 1024
+max_chunk_overlap = 20
+
+prompt_helper = PromptHelper(
+    max_input_size,
+    num_outputs,
+    max_chunk_overlap,
+)
+
+llm_predictor = LLMPredictor(
+    llm=OpenAI(temperature=0.7, model_name="text-davinci-003", max_tokens=num_outputs)
+)
+
+service_context = ServiceContext.from_defaults(
+    llm_predictor=llm_predictor, prompt_helper=prompt_helper
+)
+
+
+def construct_index(directory_path):
+    if os.path.isfile("index.json"):
+        # Index file exists, so we'll load it and add new documents to it
+        index = GPTSimpleVectorIndex.load_from_disk(
+            "index.json", service_context=service_context
+        )
+        documents = SimpleDirectoryReader(directory_path).load_data()
+        for doc in documents:
+            index.insert(doc, service_context=service_context)
+        index.save_to_disk("index.json")
+    else:
+        # Index file doesn't exist, so we'll create a new index from scratch
+        documents = SimpleDirectoryReader(directory_path).load_data()
+        index = GPTSimpleVectorIndex.from_documents(
+            documents, service_context=service_context
+        )
+        index.save_to_disk("index.json")
+
+    # Define the paths to the source and destination folders
+    absolute_path = os.path.dirname(__file__)
+    src_folder = os.path.join(absolute_path, "docs/")
+    dest_folder = os.path.join(absolute_path, "indexed_documents/")
+
+    # Get a list of all the files in the source folder
+    files = os.listdir(src_folder)
+
+
+    # Move each file from the source folder to the destination folder,
+    # except for the "do_not_delete.txt" file
+    for file in files:
+        if file != "do_not_delete.txt":
+            src_path = os.path.join(src_folder, file)
+            dest_path = os.path.join(dest_folder, file)
+            shutil.move(src_path, dest_path)
+
+    return index
 
 
 def chatbot(input_text):
@@ -24,5 +89,12 @@ def chatbot(input_text):
     return response.response, response.get_formatted_sources()
 
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface = gr.Interface(
+    fn=chatbot,
+    inputs=gr.inputs.Textbox(lines=7, label="Enter your text"),
+    outputs=[gr.Textbox(lines=30, label="Output"), gr.Textbox(lines=4, label="Source")],
+    title="Custom-trained AI Chatbot",
+)
+
+index = construct_index("docs")
+iface.launch(share=True)