gistfile1.txt

import os
from flask import Flask, request, render_template
from llama_index import GPTVectorStoreIndex, StorageContext, load_index_from_storage
from llama_index import download_loader

app = Flask(__name__)

urls=[
    'https://google.com'
]

storage = "./storage"

def init():
    global index
    if os.path.exists(storage):
        storage_context = StorageContext.from_defaults(persist_dir=storage)
        index = load_index_from_storage(storage_context)
    else:
        SimpleWebPageReader = download_loader("SimpleWebPageReader")
        loader = SimpleWebPageReader()
        documents = loader.load_data(urls=urls)
        index = GPTVectorStoreIndex.from_documents(documents)
        index.storage_context.persist(storage)

@app.route("/", methods=["GET"])
def home():
    return render_template('index.html'), 200

@app.route("/", methods=["POST"])
def form():
    global index
    query_text = request.form.get("q") 
    query_engine = index.as_query_engine()
    response = query_engine.query(str(query_text))
    return render_template('index.html', response=response, q=query_text), 200

if __name__ == "__main__":
    init()
    app.run(host="0.0.0.0", port=5000, debug=True, load_dotenv=True)