aboutsummaryrefslogtreecommitdiffstats
path: root/devourer.py
diff options
context:
space:
mode:
Diffstat (limited to 'devourer.py')
-rw-r--r--devourer.py57
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": "*",