Spaces:
Sleeping
Sleeping
from fastapi import FastAPI, Request | |
from fastapi.responses import RedirectResponse | |
from huggingface_hub import HfApi | |
from langchain_community.vectorstores.faiss import FAISS | |
from langchain_community.embeddings import OllamaEmbeddings | |
import os | |
app = FastAPI() | |
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!"} | |
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} | |
async def main2(request: Request): | |
emb = OllamaEmbeddings(model="intfloat/multilingual-e5-large-instruct", | |
base_url="https://lintasmediadanawa-hf-llm-api.hf.space", | |
embed_instruction="", | |
query_instruction="Instruct: Given a web search query, retrieve relevant passages that answer the query\nQuery: ", | |
headers={"Authorization": f"Bearer {os.getenv('HF_WRITE_TOKEN')}"}, | |
) | |
vecs = FAISS.from_texts(["haha"], emb) | |
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)} | |
async def redirect(request: Request): | |
SPACE_HOST = os.getenv("SPACE_HOST") | |
REDIRECT_URI = f"https://{SPACE_HOST}/login/callback" | |
CLIENT_ID = os.getenv("OAUTH_CLIENT_ID") | |
STATE = "test123" | |
print(SPACE_HOST, REDIRECT_URI, CLIENT_ID, STATE, "TEST PRINT") | |
url = f"https://huggingface.co/oauth/authorize?redirect_uri={REDIRECT_URI}&scope=openid%20profile&client_id={CLIENT_ID}&state={STATE}" | |
return RedirectResponse(url) | |