cleaning up backend sins now

This commit is contained in:
Andrei Stoica 2024-02-26 11:52:13 -05:00
parent 64bb9f9db3
commit eb22dd5c28
1 changed files with 6 additions and 11 deletions

View File

@ -3,6 +3,7 @@ from fastapi import FastAPI, File, Response, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse
from pydantic import BaseModel
from io import BytesIO
app = FastAPI()
@ -26,22 +27,20 @@ class Conversation(BaseModel):
@app.post("/get-text")
def stt(audio: bytes = File()):
with open("audio.webm", "wb+") as f:
f.write(audio)
async def stt(audio: bytes = File()):
with BytesIO(audio) as f:
f.name = "audio.mp3"
transcript = openAI_clinet.audio.transcriptions.create(
model="whisper-1",
file=f,
response_format="text",
)
)
data = {"len": len(audio), "user-transcript": transcript}
return data
@app.post("/conversation")
async def get_next_response(request: Request):
# role = "test"
# res_msg = "temp test response"
messages = await request.json()
res = openAI_clinet.chat.completions.create(
model="gpt-3.5-turbo",
@ -57,10 +56,6 @@ async def get_next_response(request: Request):
@app.get("/speak")
def tts(text: str):
res = openAI_clinet.audio.speech.create(
model="tts-1",
voice="nova",
input=text,
response_format='mp3'
model="tts-1", voice="nova", input=text, response_format="mp3"
)
# this works for now but I need to find a way to stream this to response
return Response(content=res.content, media_type="audio/mp3")