Update app.py

app.py

@@ -5,12 +5,18 @@ import sys
 import os
 from IPython.display import Markdown, display
 
-def  train_with_data(directory_path):
+
+def construct_index(directory_path):
+    # set maximum input size
     max_input_size = 4096
+    # set number of output tokens
     num_outputs = 2000
+    # set maximum chunk overlap
     max_chunk_overlap = 20
+    # set chunk size limit
     chunk_size_limit = 600
 
+    # define LLM
     llm_predictor = LLMPredictor(llm=OpenAI(temperature=0.5, model_name="gpt-3.5-turbo", max_tokens=num_outputs))
     prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
 
@@ -22,6 +28,9 @@ def  train_with_data(directory_path):
 
     return index
 
+
+
+
 '''
 def ask_ai():
     index = GPTSimpleVectorIndex.load_from_disk('index.json')
@@ -30,22 +39,20 @@ def ask_ai():
         response = index.query(query, response_mode="compact")
         display(Markdown(f"Response: <b>{response.response}</b>"))
 '''
-def ask_ai(query, api_key):
-    # Set default API key if user does not enter a key
-    if api_key == "":
-        api_key = "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"
-    
-    os.environ["OPENAI_API_KEY"] = api_key
-    
+# Define the ask_ai() function
+def ask_ai(question,api):
+    if api == "":
+      api = "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"
+    os.environ["OPENAI_API_KEY"] = api
     index = GPTSimpleVectorIndex.load_from_disk('index.json')
-    response = index.query(query, response_mode="compact")
+    response = index.query(question, response_mode="compact")
     return response.response
 
 
-train_with_data("data")
+construct_index("context_data")
 
 # Create Gradio interface to prompt for API key
-api_key = gr.inputs.Textbox(label="Enter your OpenAI API key (leave blank to use default key):")
+api_key = gr.inputs.Textbox(label="Enter your OpenAI API key:")
 
 # Define the interface
 iface = gr.Interface(fn=ask_ai, inputs=["text", api_key], outputs="text" ,title="Jim's Chatbot")