OwusuBlessing
112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
import os
|
|
from typing import Optional
|
|
from fastapi import FastAPI, HTTPException, Security, Depends
|
|
from fastapi.security import APIKeyHeader
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import JSONResponse
|
|
from dotenv import load_dotenv
|
|
import json
|
|
from pydantic import BaseModel
|
|
from langchain_openai import ChatOpenAI
|
|
import requests
|
|
import tempfile
|
|
from typing import Dict, Any
|
|
from fastapi.responses import Response
|
|
from datetime import datetime
|
|
from fastapi import HTTPException
|
|
from pydantic import BaseModel
|
|
from typing import Optional, Union, Dict, Any
|
|
import os
|
|
import requests
|
|
import os
|
|
from PyPDF2 import PdfReader
|
|
from scripts.transcriber import transcribe_media,group_words_into_sentences # Import the transcribe_media function
|
|
# Load environment variables
|
|
load_dotenv()
|
|
API_KEY = os.getenv("API_KEY_ACCESS")
|
|
|
|
# Initialize FastAPI app
|
|
app = FastAPI(
|
|
title="Microdot AI API",
|
|
description="API For fire fighter",
|
|
version="1.0.0"
|
|
)
|
|
|
|
# Add CORS middleware
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Setup API key authentication
|
|
api_key_header = APIKeyHeader(name="Authorization", auto_error=False)
|
|
|
|
|
|
async def get_api_key(api_key_header: str = Security(api_key_header)) -> str:
|
|
"""Validate API key from header"""
|
|
if not api_key_header or not api_key_header.startswith('Bearer '):
|
|
raise HTTPException(
|
|
status_code=401,
|
|
detail={"error": "Unauthorized", "message": "API key is missing or invalid."}
|
|
)
|
|
|
|
token = api_key_header.split(' ')[1]
|
|
if token != API_KEY:
|
|
raise HTTPException(
|
|
status_code=401,
|
|
detail={"error": "Unauthorized", "message": "API key does not match."}
|
|
)
|
|
|
|
return token
|
|
|
|
class TranscribeRequest(BaseModel):
|
|
media_url: Optional[str] = None
|
|
media_type: Optional[str] # Corrected type hint for media_type
|
|
|
|
class ChatResp(BaseModel): # Added BaseModel inheritance
|
|
error: Optional[str] = None
|
|
class TranscriptResponse(BaseModel):
|
|
transcript: dict # Changed type hint for transcript to a dictionary
|
|
|
|
@app.post("/microdot-ai/transcribe")
|
|
async def chat_endpoint(
|
|
request: TranscribeRequest,
|
|
api_key: str = Depends(get_api_key)
|
|
):
|
|
try:
|
|
|
|
# Use the transcribe_media function to transcribe the media
|
|
if request.media_url:
|
|
transcription_response = transcribe_media(request.media_url, media_type=request.media_type)
|
|
if transcription_response is None:
|
|
raise HTTPException(status_code=500, detail="Transcription failed.")
|
|
print(f"Transcription response: {transcription_response}") # Debugging print
|
|
|
|
# Parse response
|
|
words = transcription_response["results"]["channels"][0]["alternatives"][0]["words"]
|
|
transcript = group_words_into_sentences(words=words)
|
|
return TranscriptResponse(
|
|
transcript=transcript, # Corrected to return the transcript
|
|
error=None
|
|
)
|
|
|
|
except Exception as e:
|
|
print(f"Error processing chat request: {str(e)}") # Print statement added
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Error processing chat request: {str(e)}"
|
|
)
|
|
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def startup_event():
|
|
"""Initialize required components on startup"""
|
|
pass
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run("app:app", host="0.0.0.0", port=3000, reload=True) |