Update app.py

app.py

@@ -1,61 +1,31 @@
 import streamlit as st
 from llama_index import SimpleDirectoryReader, GPTVectorStoreIndex
+from langchain import OpenAI
+import gradio as gr
 import os
 import shutil
 
-
-def save_file(files):
-    directory_name = 'tmp_docs'
-
-    # Remove existing files in the directory
-    if os.path.exists(directory_name):
-        for filename in os.listdir(directory_name):
-            file_path = os.path.join(directory_name, filename)
-            try:
-                if os.path.isfile(file_path):
-                    os.remove(file_path)
-            except Exception as e:
-                print(f"Error: {e}")
-
-    # Save the new file with original filename
-    if files is not None:
-        for file in files:
-            file_name = file.name
-            file_path = os.path.join(directory_name, file_name)
-            with open(file_path, 'wb') as f:
-                shutil.copyfileobj(file, f)
-
+os.environ['OPENAI_API_KEY'] = "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"
 
 def ingest(docs_dir):
     documents = SimpleDirectoryReader(docs_dir).load_data()
     index = GPTVectorStoreIndex.from_documents(documents)
     return index
 
-
-def get_answer(index, message):
-    response = query(index, message)
-    return [('Chatbot', ''.join(response.response))]
-
-
 def query(index, query_text):
     query_engine = index.as_query_engine()
     response = query_engine.query(query_text)
     return response
-
-
-os.environ['OPENAI_API_KEY'] = st.text_input(
-    "Enter your OpenAI API key", type="password")
+    
+def get_answer(index, message):
+    response = query(index, message)
+    return [('Chatbot', ''.join(response.response))]
 
 if os.environ['OPENAI_API_KEY']:
     # Initialize chatbot history
     chatbot = []
 
-    # Display file upload component
-    files = st.file_uploader('Upload Files', accept_multiple_files=True)
-    if files is not None:
-        save_file(files)
-
-    index = ingest('tmp_docs')
+    index = ingest('temp_docs')
 
     # Display message input component
     message = st.text_input('Enter message')