Update app.py

app.py

@@ -1,43 +1,19 @@
 import json
-from langchain.chains import llm
-from ast import Index
-from gpt_index import SimpleDirectoryReader, GPTListIndex, readers, GPTSimpleVectorIndex, LLMPredictor, PromptHelper
-from langchain import OpenAI
-import openai
-import os 
 import streamlit as st
+from gpt_index import GPTSimpleVectorIndex
 
-# Load the GPT index from the file
-with open('index-4.json', 'r') as f:
-    index_data = json.load(f)
-gpt_index = GPTSimpleVectorIndex(index_data)
+def search_gpt_index():
+    # Load the GPT index from the file
+    with open('index.json', 'r') as f:
+        index_data = json.load(f)
+    gpt_index = GPTSimpleVectorIndex(index_data)
 
-OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
-os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+    # Define the app layout
+    st.title('GPT Search')
+    query = st.text_input('Enter a query:')
+    if query:
+        result = gpt_index.query(query)
+        st.write(result)
 
-def ask_patrick():
-    # Get the user's query
-    query = st.text_input("Tell me about Patrick Collison?")
-    
-    # Use OpenAI to get a response to the query
-    response = openai.Completion.create(
-        engine="text-davinci-002",
-        prompt=f"Patrick Collison is a tech entrepreneur. {query}",
-        temperature=0.5,
-        max_tokens=1024,
-        top_p=1,
-        frequency_penalty=0,
-        presence_penalty=0
-    )
-
-    # Print the response
-    st.write(response.choices[0].text)
-    
-    # Search the GPT index for the best response
-    result = gpt_index.query(response.choices[0].text)
-    
-    # Print the result
-    st.write(result)
-
-if __name__ == "__main__":
-    ask_patrick()
+if __name__ == '__main__':
+    search_gpt_index()