Rewritten to use TailwindCSS and HTMX 🔥

app/app.py

@@ -1,68 +1,90 @@
-import logging
-import os
-from flask import Flask, render_template, Response, request, session
-from main import yoink, process, user_streams, stream_lock
-import uuid  # Import UUID
-
-app = Flask(__name__, static_folder="website/static", template_folder="website")
-app.secret_key = os.urandom(24)  # Necessary for using sessions
-
-
-# Configure logging
-logging.basicConfig(
-    filename='./logs/app.log',
-    level=logging.DEBUG,
-    format='%(asctime)s %(levelname)s: %(message)s',
-    datefmt='%Y-%m-%d %H:%M:%S'
-)
-
-def create_session():
-    session_id = str(uuid.uuid4())
-    # This should never happen but I'm putting the logic there anyways
-    try:
-        if user_streams[session_id]:
-            session_id = create_session()
-    except KeyError:
-        pass
-    return session_id
-    
-
-@app.route('/')
-def home():
-    session_id = create_session()
-    session['id'] = session_id
-    logging.info(f"Home page accessed. Assigned initial session ID: {session_id}")
-    return render_template('index.html', session_id=session_id)
-
-@app.route('/process_url', methods=['POST'])
-def process_url():
-    session_id = session.get('id')
-    if not session_id:
-        session_id = create_session()
-        session['id'] = session_id
-        logging.info(f"No existing session. Created new session ID: {session_id}")
-
-    url = request.form['url']
-    logging.info(f"Received URL for processing from session {session_id}: {url}")
-    success, msg, status_code, = process(url, session_id)
-
-    if success:
-        logging.info(f"Processing started successfully for session {session_id}.")
-        return Response("Processing started. Check /stream_output for updates.", content_type='text/plain', status=200)
-    else:
-        logging.error(f"Processing failed for session {session_id}: {msg}")
-        return Response(msg, content_type='text/plain', status=status_code)
-
-@app.route('/stream_output')
-def stream_output():
-    session_id = session.get('id')
-    if not session_id or session_id not in user_streams:
-        logging.warning(f"Stream requested without a valid session ID: {session_id}")
-        return Response("No active stream for this session.", content_type='text/plain', status=400)
-    logging.info(f"Streaming output requested for session {session_id}.")
-    return Response(yoink(session_id), content_type='text/plain', status=200)
-
-if __name__ == '__main__':
-        logging.info("Starting Flask application.")
-        app.run(debug=True, threaded=True)  # Enable threaded to handle multiple requests
-    
+import logging
+import os
+import uuid
+from flask import Flask, render_template, Response, request, session
+from main import yoink, process, user_streams, stream_lock
+
+app = Flask(__name__, static_folder="website/static", template_folder="website")
+app.secret_key = os.urandom(24)  # Necessary for using sessions
+
+# Configure logging
+logging.basicConfig(
+    filename='./logs/app.log',
+    level=logging.DEBUG,
+    format='%(asctime)s %(levelname)s: %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S'
+)
+
+def create_session():
+    """
+    Create a new session by generating a UUID and ensuring it does not collide 
+    with an existing session in the user_streams global dictionary.
+    
+    Returns:
+        str: A unique session ID.
+    """
+    session_id = str(uuid.uuid4())
+    # Even though collisions are unlikely, we check for safety.
+    try:
+        if user_streams[session_id]:
+            session_id = create_session()
+    except KeyError:
+        pass
+    return session_id
+
+@app.route('/')
+def home():
+    """
+    Render the home page and initialize a session.
+    
+    Returns:
+        Response: The rendered home page with a unique session id.
+    """
+    session_id = create_session()
+    session['id'] = session_id
+    logging.info(f"Home page accessed. Assigned initial session ID: {session_id}")
+    return render_template('index.html', session_id=session_id)
+
+@app.route('/process_url', methods=['POST'])
+def process_url():
+    """
+    Accept a YouTube URL (from a form submission), initialize the session if necessary, 
+    and trigger the transcript retrieval and AI processing.
+    
+    Returns:
+        Response: Text response indicating start or error message.
+    """
+    session_id = session.get('id')
+    if not session_id:
+        session_id = create_session()
+        session['id'] = session_id
+        logging.info(f"No existing session. Created new session ID: {session_id}")
+    url = request.form['url']
+    logging.info(f"Received URL for processing from session {session_id}: {url}")
+    success, msg, status_code = process(url, session_id)
+    if success:
+        logging.info(f"Processing started successfully for session {session_id}.")
+        return Response("Processing started. Check /stream_output for updates.", content_type='text/plain', status=200)
+    else:
+        logging.error(f"Processing failed for session {session_id}: {msg}")
+        return Response(msg, content_type='text/plain', status=status_code)
+
+@app.route('/stream_output')
+def stream_output():
+    """
+    Stream the AI processing output for the current session.
+    
+    Returns:
+        Response: A streaming response with text/plain content.
+    """
+    session_id = session.get('id')
+    if not session_id or session_id not in user_streams:
+        logging.warning(f"Stream requested without a valid session ID: {session_id}")
+        return Response("No active stream for this session.", content_type='text/plain', status=400)
+    logging.info(f"Streaming output requested for session {session_id}.")
+    return Response(yoink(session_id), content_type='text/plain', status=200)
+
+if __name__ == '__main__':
+    logging.info("Starting Flask application.")
+    # Running with threaded=True to handle multiple requests concurrently.
+    app.run(debug=True, threaded=True)