File size: 1,828 Bytes
e95de35 eb66dcb eed80a2 e95de35 ef93b5e eb66dcb eed80a2 eb66dcb ef93b5e eb66dcb ef93b5e eb66dcb ef93b5e eb66dcb ef93b5e eb66dcb ef93b5e 5916e02 eb66dcb e95de35 ef93b5e e95de35 ef93b5e e95de35 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
from fastapi import FastAPI, HTTPException,UploadFile,File
from pydantic import BaseModel
from deep_translator import GoogleTranslator
from fastapi.responses import JSONResponse
import os
from main import process,audio_process
from dotenv import load_dotenv
load_dotenv()
# Create the FastAPI app instance
os.makedirs("/tmp/huggingface_cache", exist_ok=True)
os.environ["HF_HOME"] = "/tmp/huggingface_cache"
app = FastAPI()
# Root endpoint
@app.get("/")
async def home():
return {"message": "Testing_api"}
# Token verification function
def verify_token(token: str):
if token != os.getenv("TOKEN"):
raise HTTPException(status_code=401, detail="Token not matched")
# Translate endpoint that accepts a query parameter 'text'
@app.get("/translate")
async def translate(text: str = "", token: str = ""):
if not text or not token:
raise HTTPException(status_code=400, detail="No text or token provided")
verify_token(token)
translator = GoogleTranslator(source="auto", target="mr")
result = translator.translate(text)
return {"result": result}
@app.get("/chatbot")
async def chatbot(text: str = "", token: str = ""):
if not text or not token:
raise HTTPException(status_code=400, detail="No text provided")
verify_token(token)
result = process(user_query=text)
return {"result": result}
@app.post("/audio_chat")
async def audio_chat(audio: UploadFile = File(...), token: str = ""):
if not audio or not token:
raise HTTPException(status_code=400, detail="No audio file provided")
verify_token(token)
try:
result = audio_process(audio.file) # Replace with actual audio processing logic
return {"result": result}
except Exception as e:
raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
|