app.py

from flask import Flask, send_from_directory, request, jsonify
from flask_cors import CORS
from llm.justin_embeddings import init_everything
from dotenv import load_dotenv
import threading
import os
from langchain.llms import OpenAI
from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain, SequentialChain 
from langchain.memory import ConversationBufferMemory
from llm.query import *

app = Flask(__name__, static_folder='client/build', static_url_path='')

load_dotenv()

#prompt templates
description_template = PromptTemplate(
    input_variables = ['description'],
    template = 'act as prospective real state buyer. Here is little bit of information about you : {description}. Write a small 200 word paragraph describing your ideal home'
)
#memory
description_memory = ConversationBufferMemory(input_key='description', memory_key='chat_history')

#llms
llm = OpenAI(temperature=0.1)
description_chain = LLMChain(llm=llm, prompt= description_template,verbose=True, output_key='description', memory=description_memory)
# form_response_chain = LLMChain(llm=llm)

# Enable CORS for all routes
CORS(app)

# init the vector DB in a separate thread
def run_in_background():
    print("=== Call init everything")
    # init_everything()

import json

def load_data(filename):
    """Loads JSON data from the specified file."""
    with open(filename, 'r') as f:
        return json.load(f)

@app.route('/api/prompt', methods=['POST'])
def handle_prompt():
    
    data = request.json
    prompt_text = data.get('prompt', '') 
    form_values = data.get('formValues', '')

    print("Here are the form values:")
    print(form_values)

    # prompt_response = 
    desc = description_chain.run(prompt_text)

    if form_values:
        min_price = form_values['minPrice']
        max_price = form_values['maxPrice']
    else:
        min_price = 0
        max_price = 1000000

    dictionary = get_json(query=desc, min_price=min_price, max_price=max_price)
    dictionary['text-response'] = 'Heres what you might be looking for'
    return jsonify(dictionary)

@app.route('/')
def index():
    return send_from_directory(app.static_folder, 'index.html')

if __name__ == "__main__":
    threading.Thread(target=run_in_background).start()
    app.run(debug=True)