from fastapi import FastAPI, Request
from huggingface_hub import HfApi
from langchain_community.vectorstores.faiss import FAISS


import os

app = FastAPI()

@app.get("/")
def greet_json():
    with open("test.txt", "w") as f:
        f.write("HELLO WORLD")
    api = HfApi()

    api.upload_file(
    
        path_or_fileobj="test.txt",
    
        path_in_repo="test.txt",
    
        repo_id="jonathanjordan21/test-dataset",
    
        repo_type="dataset",

        token = os.getenv("HF_WRITE_TOKEN")
    
    )
    return {"Hello": "World!"}


@app.get("/{rand_int}/")
async def main(rand_int: str, request: Request):
    with open("test.txt", "w") as f:
        f.write(rand_int)
    api = HfApi()

    api.upload_file(
    
        path_or_fileobj="test.txt",
    
        path_in_repo="test.txt",
    
        repo_id="jonathanjordan21/test-dataset",
    
        repo_type="dataset",

        token = os.getenv("HF_WRITE_TOKEN")
    
    )
    return {"raw_url": str(request.url), "message":rand_int}




@app.get("/vecs/")
async def main2(request: Request):
    vecs = FAISS.from_texts(["haha"])
    api = HfApi()

    vecs.save_local("faiss_index")

    api.upload_folder(

        folder_path="faiss_index",
    
        repo_id="jonathanjordan21/test-dataset",

        path_in_repo="vecs/faiss_index",
    
        repo_type="dataset",

        token = os.getenv("HF_WRITE_TOKEN")
    
    )

    return {"raw_url": str(request.url)}