from flask import Flask, render_template, request
from htmx_flask import Htmx
import random

from autoAidModules.search_funcs import *

htmx = Htmx()

app = Flask(__name__)

current_task = []

my_cars = {
    "2005 Subaru Forester XT": 0,
    "2001 BMW 540i": 0,
}

selected_car = "2001 BMW 540i"

htmx.init_app(app)

@app.route("/")
def hello_world():
    for _ in range(len(current_task)):
        current_task.pop()
    return render_template("index.html",my_cars=my_cars.items(), selected=selected_car)

@app.route("/vehicles")
def vehicles():
    if request.headers.get("HX-Request"):
        return render_template("vehicles_list.html", my_cars=my_cars.items(), selected=selected_car)
    return "Not an HX request"

@app.route("/new_vehicle", methods=["POST"])
def new_vehicle():
    if request.headers.get("HX-Request"):
        my_cars[request.form["newVEHICLE"]] = 0
        return render_template("vehicles_list.html", my_cars=my_cars.items(), selected=selected_car)
    return "Not an HX request"

@app.route("/search", methods=["POST"])
def search():
    if request.headers.get("HX-Request"):
        print(request.form["search"])
        search_query = request.form["search"].title()
        print(request.form["car_details"])
        make = request.form["car_details"].split(" ")[1]
        car_details = request.form["car_details"]
        for _ in range(len(current_task)):
            current_task.pop()
        current_task.append({
            "date": "Oct 7",
            "desc_plain": "Searching for ",
            "desc_bold": search_query,
            "icon": "check"
        })

        pref_forum = get_preferred_forums(make)
        current_task.append({
            "date": "Oct 7",
            "desc_plain": "Searching for forums related to ",
            "desc_bold": pref_forum[0],
            "icon": "prog"
        })
        data = search_on_forum(pref_forum[0], search_query, random.randint(4,8))
        current_task[-1]["icon"] = "check"
        current_task.append({
            "date": "Oct 7",
            "desc_plain": "Found the following results: ",
            "desc_bold": f"{len(data)} forum pages",
            "icon": "check"
        })
        current_task.append({   
            "date": "Oct 7",
            "desc_plain": "Using Claude to generate a tasklist",
            "desc_bold": "",
            "icon": "prog"
        })
        pred = get_tasks_from_pages(data, search_query, car_details)
        current_task[-1]["icon"] = "check"
        tasks = ["Have you tried turning your car on and off?"]
        try:
            tasks = pred.split('\n\n')[0].replace("- ","").splitlines()
            my_cars[car_details] += 1
            print(my_cars, "MY CARS")
        except:
            print("Uh oh! Claude didn't return any results!")
        return render_template("tasks.html", search_query=search_query, to_do=tasks)
    
@app.route("/progress")
def progress():
    if request.headers.get("HX-Request"):
        print(f"current ask is", current_task)
        return render_template("progress.html", tasks=current_task)

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