main.py

import os
from flask import Flask, render_template, request, session
from dotenv import load_dotenv
load_dotenv()

from embedchain import App
chat_bot_app = App()
bot_name="Naval Ravikant"

app = Flask(__name__)
app.static_folder = 'static'
app.secret_key = 'embedchain'

def load_app():
    chat_bot_app.add("youtube_video", "https://www.youtube.com/watch?v=3qHkcs3kG44")
    chat_bot_app.add("pdf_file", "https://navalmanack.s3.amazonaws.com/Eric-Jorgenson_The-Almanack-of-Naval-Ravikant_Final.pdf",)
    chat_bot_app.add("web_page", "https://nav.al/feedback")
    chat_bot_app.add("web_page", "https://nav.al/agi")
    chat_bot_app.add_local("qna_pair", ("Who is Naval Ravikant?", "Naval Ravikant is an Indian-American entrepreneur and investor."))


@app.route("/", methods=["GET", "POST"])
def home():
    context = {}
    context["bot_name"] = bot_name
    context["chats"] = session.get('chats', [])
    if request.method == "POST":
        user_query = request.form["query"]
        context["chats"].append(("H", user_query))

        query_result = chat_bot_app.chat(user_query)
        context["chats"].append(("B", query_result))

        session['chats'] = context["chats"]
    return render_template("index.html", **context)


if __name__ == "__main__":
    load_app()
    app.run(debug=False, port=8000)