Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-import openai 
+import openai
 import gradio as gr
 import chroma
 
@@ -20,6 +20,12 @@ def create_embeddings(text):
 def save_embeddings(embeddings):
     database.insert_embeddings(embeddings)
 
+# Function to load the source document
+def load_source_document():
+    with open("source.txt", "r") as f:
+        text = f.read()
+    return text
+
 # Function to create a chatbot
 def create_chatbot():
     engine = gr.engine.Engine(title="Chatbot")
@@ -41,6 +47,13 @@ def create_chatbot():
 
         return response["choices"][0]["text"]
 
+    # Load the source document
+    text = load_source_document()
+
+    # Create embeddings from the source document and save them in the Chroma database
+    embeddings = create_embeddings(text)
+    save_embeddings(embeddings)
+
     # Create a button to start the chatbot
     button = gr.Button(label="Start Chatting", description="Click to start chatting", on_click=handle_input)
 
@@ -53,4 +66,4 @@ def create_chatbot():
 chatbot = create_chatbot()
 
 # Run the chatbot
-chatbot.launch()
+chatbot.launch()