from fastapi import FastAPI
import os
from fastapi.responses import StreamingResponse, FileResponse
from io import BytesIO
from claude_api import Client
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.get("/prompt/")
async def get_prompt(prompt: str = "Hello World"):
    """
    """
    cookie = os.environ.get('CLAUDE_TOKEN', None)
    claude_api = Client(cookie)
    new_chat = claude_api.create_new_chat()
    conversation_id = new_chat['uuid']
    response = claude_api.send_message(prompt, conversation_id)
    return all_files