Create app.py

app.py

@@ -0,0 +1,35 @@
+import logging
+import sys
+import gradio as gr
+from llama_index import GPTVectorStoreIndex, SimpleDirectoryReader, ServiceContext
+from llama_index import OpenAIEmbedding
+
+logging.basicConfig(stream=sys.stdout, level=logging.INFO)
+logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
+os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
+
+
+def load_data():
+    return SimpleDirectoryReader("./document/management/").load_data()
+
+def build_index(documents):
+    return GPTVectorStoreIndex.from_documents(documents)
+
+def query_index(index, similarity_top_k=3, streaming=True):
+    query_engine = index.as_query_engine(
+        similarity_top_k=similarity_top_k,
+        streaming=streaming
+    )
+    return query_engine  # return the query engine instance
+
+data = load_data()
+index = build_index(data)
+query_engine = query_index(index)  # initialize the query engine
+
+def get_response(text,history=None):
+    # Use the initialized query engine to perform the query
+    response = str(query_engine.query(text))
+    return response
+
+t = gr.ChatInterface(get_response, analytics_enabled=True)
+t.launch(debug=True, share=True)