import os
import time
import sqlite3
import threading
from pydantic import BaseModel
from fastapi import FastAPI
from fastapi.responses import HTMLResponse
from huggingface_hub import HfApi
from filesystem import upload, download

app = FastAPI()

class sql(BaseModel):
    sql: str

download()

def Upload():
    while True:
        upload()
        time.sleep(30)

def Connect():
    conn = sqlite3.connect("database.db")
    cursor = conn.cursor()
    return conn, cursor

def Close(conn: sqlite3.Connection):
    conn.commit()
    conn.close()

thread = threading.Thread(target=Upload)
thread.start()

@app.get("/", response_class=HTMLResponse)
def HuggingFaceSpaceRoot():
    return ""

@app.head("/")
def UptimeRobot():
    return "OK"

@app.post("/sql")
def execute_sql(param: sql):
    conn, cursor = Connect()
    cursor.execute(param.sql)
    Close(conn)

@app.get("/sql")
def execute_sql_get(sql: str):
    conn, cursor = Connect()
    cursor.execute(sql)
    Close(conn)