diff options
| -rw-r--r-- | devourer.py | 57 |
1 files changed, 24 insertions, 33 deletions
diff --git a/devourer.py b/devourer.py index 8532bf3..0a64e8e 100644 --- a/devourer.py +++ b/devourer.py @@ -16,6 +16,7 @@ from readability import Document from gtts import gTTS from datetime import datetime as time from fastapi import FastAPI +from fastapi import Response as APIResponse # FIXME-maybe actually really do some logging @@ -271,18 +272,17 @@ nltk.download("punkt") @app.get("/mila/tika") -async def pdf_to_audio_ep(url: str): +def pdf_to_audio_ep(url: str): """turns a pdf into an audiofile""" audio_path = pdfToVoice() - return { - "Content-Type": "application/json", - "isOK": True if audio_path != "" else False, - "audio": getAudioFromFile(audio_path) if audio_path != "" else "", - } + return APIResponse( + getAudioFromFile(audio_path) if audio_path != "" else "", + media_type="audio/mpeg", + ) @app.get("/mila/reqs") -async def extract_reqs_ep(url: str, sourcetype: str = "html"): +def extract_reqs_ep(url: str, sourcetype: str = "html"): """extracts the requirements from a given url""" result = getRequirements() return { @@ -293,18 +293,15 @@ async def extract_reqs_ep(url: str, sourcetype: str = "html"): @app.get("/mila/wiki") -async def wiki_search_ep(term: str, audio: bool = False): +def wiki_search_ep(term: str, audio: bool = False): """search and summarizes from wikipedia""" text = searchWikipedia(term) if audio: audio_path = textToAudio(text) - return { - "Content-Type": "application/json", - "isOK": (True if audio_path != "" else False) - and (True if text != "" else False), - "audio": getAudioFromFile(audio_path) if audio_path != "" else "", - "text": text, - } + return FastAPI( + getAudioFromFile(audio_path) if audio_path != "" else "", + media_type="audio/mpeg", + ) else: return { "Content-Type": "application/json", @@ -315,19 +312,16 @@ async def wiki_search_ep(term: str, audio: bool = False): @app.get("/mila/summ") -async def summarize_ep(url: str, summary: str = "none", audio: bool = False): +def summarize_ep(url: str, summary: str = "none", audio: bool = False): """summarize and turn the summary into audio""" text = summarizeLinkToAudio(url, summary) if audio: audio_path = textToAudio(text) print(audio_path) - return { - "Content-Type": "audio/mpeg", - # "isOK": (True if audio_path != "" else False) - # and (True if text != "" else False), - "audio": getAudioFromFile(audio_path) if audio_path != "" else "", - # "text": text, - } + return APIResponse( + getAudioFromFile(audio_path) if audio_path != "" else "", + media_type="audio/mpeg", + ) else: return { "Content-Type": "application/json", @@ -338,19 +332,16 @@ async def summarize_ep(url: str, summary: str = "none", audio: bool = False): @app.get("/mila/mila") -async def mila_ep(url: str, summary: str = "newspaper", audio: bool = False): +def mila_ep(url: str, summary: str = "newspaper", audio: bool = False): """extract all the urls and then summarize and turn into audio""" text = summarizeLinksToAudio(url, summary) if audio: audio_path = textToAudio(text) print(audio_path) - return { - "Content-Type": "application/json", - "isOK": (True if audio_path != "" else False) - and (True if text != "" else False), - "audio": getAudioFromFile(audio_path) if audio_path != "" else "", - "text": text, - } + return APIResponse( + getAudioFromFile(audio_path) if audio_path != "" else "", + media_type="audio/mpeg", + ) else: return { "Content-Type": "application/json", @@ -361,12 +352,12 @@ async def mila_ep(url: str, summary: str = "newspaper", audio: bool = False): @app.get("/mila/health") -async def health_ep(): +def health_ep(): return {"isOK": True} @app.get("/mila/robots.txt") -async def robots_ep(): +def robots_ep(): return { "Content-Type": "apllication/json", "User-Agents": "*", |