foreverpyrite/screw-bardo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

i read PEP conventions

Browse Source
This commit is contained in:
ForeverPyrite
2024-11-20 15:50:41 -05:00
parent 0696ed21e9
commit c363f4c246
1 changed files with 34 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
app/app.py
View File

@@ -4,8 +4,6 @@ from asyncio import sleep
from datetime import datetime from datetime import datetime
import threading, pytz import threading, pytz
app = Flask(__name__, static_folder="website/static", template_folder="website") app = Flask(__name__, static_folder="website/static", template_folder="website")
@app.route('/') @app.route('/')
@@ -14,49 +12,51 @@ def home():
@app.route('/process_url', methods=['POST']) @app.route('/process_url', methods=['POST'])
def process_url(): def process_url():
global thread url = request.form.get('url', '').strip()
log(f"\n\n\n## New Entry at {datetime.now(pytz.timezone('America/New_York')).strftime('%Y-%m-%d %H:%M:%S')}\n\n") if not url:
url = request.form['url'] log("No URL provided.\n")
return "No URL provided.", 400
log(f"\n\n\n## New Entry at {datetime.now(pytz.timezone('America/New_York')).strftime('%Y-%m-%d %H:%M:%S')}\n")
log(f"URL: {url}\n") log(f"URL: {url}\n")
# Extract the video ID from the URL # Extract the video ID from the URL
video_id = get_video_id(url) # Modify this function to accept the URL video_id = get_video_id(url)
if not video_id: if not video_id:
log(f"Could not parse video id from URL: {url}") log(f"Could not parse video id from URL: {url}\n")
return "Couldn't parse video ID from URL. (Are you sure you entered a valid YouTube.com or YouTu.be URL?)" return "Couldn't parse video ID from URL. (Ensure it's a valid YouTube.com or YouTu.be URL.)", 400
log(f"Video ID: {video_id}\n\n") log(f"Video ID: {video_id}\n")
# Get the transcript for that video ID # Get the transcript for that video ID
transcript = get_auto_transcript(video_id) transcript = get_auto_transcript(video_id)
if (not transcript): if not transcript:
log("## Error: could not retrieve transcript, Assistant won't be called.") log("## Error: Could not retrieve transcript. Assistant won't be called.\n")
return "Successfully parsed video ID from URL, however the ID was either invalid, the transcript was disabled by the video owner, or some other error was raised because of YouTube." return "Parsed video ID, but transcript retrieval failed (might be disabled by the video owner).", 400
thread = threading.Thread(name="create_stream", target=create_and_stream, args=(transcript,)) # The comma here is very intentional, it's so that it iterates it as a tuple rather than iterateing the string. # Start the stream thread
log("Stream preperation complete, sending reply...\n\n") thread = threading.Thread(target=create_and_stream, args=(transcript,))
return Response("Processing started. Check /stream_output for updates.", content_type='text/plain', status=200) # Add more detailed output if needed thread.start()
log("Stream preparation complete, sending reply...\n")
return Response("Processing started. Check /stream_output for updates.", content_type='text/plain', status=200)
@app.route('/stream_output') @app.route('/stream_output')
def stream_output(): def stream_output_route():
def yoink(): def generate():
log("<details>\n<summary>Starting stream thread...</summary>\n\n") log("<details>\n<summary>Starting stream thread...</summary>\n\n")
thread.start()
# Start streaming output from output_stream # Start streaming output from output_stream
log("Starting to stream output.") log("Starting to stream output.\n")
while not output_stream.done: while not output_stream.done:
if output_stream.buffer != []: if output_stream.buffer:
delta = output_stream.buffer.pop(0) delta = output_stream.buffer.pop(0)
yield bytes(delta, encoding="utf-8") yield delta.encode("utf-8")
else: else:
awaiter(sleep(0.05)) awaiter(sleep(0.05))
log(f"\nStream successfully completely.\n\n</details>\n\n---\n\n### Completed Assistant Response:\n{output_stream.response}\n\n---\n\n") log(f"\nStream completed.\n</details>\n\n---\n\n### Completed Assistant Response:\n{output_stream.response}\n\n---\n\n")
output_stream.reset() output_stream.reset()
thread.join() log("\n### Task completed successfully without errors!\n")
log("\n### Task completed sucessfully without errors!")
return
return Response(yoink(), content_type='text/plain', status=200)
return Response(generate(), content_type='text/plain', status=200)
if __name__ == '__main__': if __name__ == '__main__':
app.run(debug=True) app.run(debug=True, host='0.0.0.0', port=1986)