Update app.py

app.py

@@ -1,135 +1,135 @@
-import os
-from flask import Flask, render_template, request, jsonify, send_from_directory
-from flask_cors import CORS
-from werkzeug.utils import secure_filename
-from pdf_logic import process_comparison
-
-app = Flask(__name__)
-CORS(app)
-
-# Configuration
-UPLOAD_FOLDER = 'static/uploads'
-OUTPUT_FOLDER = 'static/output'
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
-os.makedirs(OUTPUT_FOLDER, exist_ok=True)
-
-
-@app.route('/compare', methods=['POST'])
-def compare():
-    if 'source' not in request.files or 'destination' not in request.files:
-        return jsonify({'error': 'Missing files'}), 400
-    
-    source_file = request.files['source']
-    dest_file = request.files['destination']
-    
-    s_path = os.path.join(UPLOAD_FOLDER, secure_filename(source_file.filename))
-    d_path = os.path.join(UPLOAD_FOLDER, secure_filename(dest_file.filename))
-    
-    source_file.save(s_path)
-    dest_file.save(d_path)
-    
-    try:
-        # Run the comparison logic
-        src_out, dest_out, changes, total_pages = process_comparison(s_path, d_path, OUTPUT_FOLDER)
-        
-        return jsonify({
-            'status': 'success',
-            'source_pdf': f"/static/output/{src_out}",
-            'dest_pdf': f"/static/output/{dest_out}",
-            'changes': changes,
-            'total_pages': total_pages
-        })
-    except Exception as e:
-        print(e)
-        return jsonify({'error': str(e)}), 500
-
-# WorkOS Configuration
-import workos
-from workos import WorkOSClient
-
-# Initialize WorkOS Client
-# In production, use environment variables: os.environ.get('WORKOS_API_KEY')
-# For now we will check if they exist, otherwise these endpoints will fail gracefully or you can hardcode for testing if needed
-workos.api_key = os.environ.get('WORKOS_API_KEY', 'placeholder_key') 
-workos.client_id = os.environ.get('WORKOS_CLIENT_ID', 'placeholder_id')
-
-@app.route('/auth/verify-session', methods=['POST'])
-def verify_session():
-    """
-    Enforces single-session policy.
-    Expects JSON: { "userId": "user_...", "currentSessionId": "session_..." }
-    """
-    data = request.json
-    user_id = data.get('userId')
-    current_session_id = data.get('currentSessionId')
-
-    if not user_id or not current_session_id:
-        return jsonify({'error': 'Missing userId or currentSessionId'}), 400
-
-    try:
-        # 1. List all active sessions for the user
-        # Note: WorkOS Python SDK ListSessions usage might vary by version; 
-        # checking User Management > List Sessions or similar via SDK.
-        # As of recent versions, it's workos.user_management.list_sessions(user_id=...)
-        # or verify standard usage. We'll use the generic client resources if specific helper isn't found,
-        # but workos.user_management is the standard path.
-        
-        sessions_list = workos.user_management.list_sessions(
-            user_id=user_id,
-            type='user_session' # Ensure we get user sessions
-        )
-        
-        active_sessions = sessions_list.data
-        revoked_count = 0
-
-        # 2. Iterate and revoke any session that is NOT the current one
-        for session in active_sessions:
-            if session.id != current_session_id:
-                # Revoke this old/extra session
-                workos.user_management.revoke_session(session.id)
-                revoked_count += 1
-
-        return jsonify({
-            'status': 'success',
-            'revoked_count': revoked_count,
-            'message': f'Session verified. {revoked_count} other sessions revoked.'
-        })
-
-    except Exception as e:
-        print(f"WorkOS Error: {e}")
-        # If API key is invalid or not set, this will error.
-        # We return 500 but frontend should handle it (maybe allow login if strict mode is off)
-        return jsonify({'error': str(e)}), 500
-
-
-@app.route('/auth/logout-everywhere', methods=['POST'])
-def logout_everywhere():
-    """
-    Revokes ALL sessions for the user.
-    Expects JSON: { "userId": "user_..." }
-    """
-    data = request.json
-    user_id = data.get('userId')
-
-    if not user_id:
-        return jsonify({'error': 'Missing userId'}), 400
-
-    try:
-        # List all sessions
-        sessions_list = workos.user_management.list_sessions(user_id=user_id)
-        
-        # Revoke all
-        for session in sessions_list.data:
-            workos.user_management.revoke_session(session.id)
-            
-        return jsonify({'status': 'success', 'message': 'All sessions revoked.'})
-
-    except Exception as e:
-        print(f"WorkOS Error: {e}")
-        return jsonify({'error': str(e)}), 500
-
-if __name__ == '__main__':
-    app.run(port=5000)
-    # from waitress import serve
-
-    # serve(app, host="0.0.0.0", port=7860)
+import os
+from flask import Flask, render_template, request, jsonify, send_from_directory
+from flask_cors import CORS
+from werkzeug.utils import secure_filename
+from pdf_logic import process_comparison
+
+app = Flask(__name__)
+CORS(app)
+
+# Configuration
+UPLOAD_FOLDER = 'static/uploads'
+OUTPUT_FOLDER = 'static/output'
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+os.makedirs(OUTPUT_FOLDER, exist_ok=True)
+
+
+@app.route('/compare', methods=['POST'])
+def compare():
+    if 'source' not in request.files or 'destination' not in request.files:
+        return jsonify({'error': 'Missing files'}), 400
+    
+    source_file = request.files['source']
+    dest_file = request.files['destination']
+    
+    s_path = os.path.join(UPLOAD_FOLDER, secure_filename(source_file.filename))
+    d_path = os.path.join(UPLOAD_FOLDER, secure_filename(dest_file.filename))
+    
+    source_file.save(s_path)
+    dest_file.save(d_path)
+    
+    try:
+        # Run the comparison logic
+        src_out, dest_out, changes, total_pages = process_comparison(s_path, d_path, OUTPUT_FOLDER)
+        
+        return jsonify({
+            'status': 'success',
+            'source_pdf': f"/static/output/{src_out}",
+            'dest_pdf': f"/static/output/{dest_out}",
+            'changes': changes,
+            'total_pages': total_pages
+        })
+    except Exception as e:
+        print(e)
+        return jsonify({'error': str(e)}), 500
+
+# WorkOS Configuration
+import workos
+from workos import WorkOSClient
+
+# Initialize WorkOS Client
+# In production, use environment variables: os.environ.get('WORKOS_API_KEY')
+# For now we will check if they exist, otherwise these endpoints will fail gracefully or you can hardcode for testing if needed
+workos.api_key = os.environ.get('WORKOS_API_KEY', 'placeholder_key') 
+workos.client_id = os.environ.get('WORKOS_CLIENT_ID', 'placeholder_id')
+
+@app.route('/auth/verify-session', methods=['POST'])
+def verify_session():
+    """
+    Enforces single-session policy.
+    Expects JSON: { "userId": "user_...", "currentSessionId": "session_..." }
+    """
+    data = request.json
+    user_id = data.get('userId')
+    current_session_id = data.get('currentSessionId')
+
+    if not user_id or not current_session_id:
+        return jsonify({'error': 'Missing userId or currentSessionId'}), 400
+
+    try:
+        # 1. List all active sessions for the user
+        # Note: WorkOS Python SDK ListSessions usage might vary by version; 
+        # checking User Management > List Sessions or similar via SDK.
+        # As of recent versions, it's workos.user_management.list_sessions(user_id=...)
+        # or verify standard usage. We'll use the generic client resources if specific helper isn't found,
+        # but workos.user_management is the standard path.
+        
+        sessions_list = workos.user_management.list_sessions(
+            user_id=user_id,
+            type='user_session' # Ensure we get user sessions
+        )
+        
+        active_sessions = sessions_list.data
+        revoked_count = 0
+
+        # 2. Iterate and revoke any session that is NOT the current one
+        for session in active_sessions:
+            if session.id != current_session_id:
+                # Revoke this old/extra session
+                workos.user_management.revoke_session(session.id)
+                revoked_count += 1
+
+        return jsonify({
+            'status': 'success',
+            'revoked_count': revoked_count,
+            'message': f'Session verified. {revoked_count} other sessions revoked.'
+        })
+
+    except Exception as e:
+        print(f"WorkOS Error: {e}")
+        # If API key is invalid or not set, this will error.
+        # We return 500 but frontend should handle it (maybe allow login if strict mode is off)
+        return jsonify({'error': str(e)}), 500
+
+
+@app.route('/auth/logout-everywhere', methods=['POST'])
+def logout_everywhere():
+    """
+    Revokes ALL sessions for the user.
+    Expects JSON: { "userId": "user_..." }
+    """
+    data = request.json
+    user_id = data.get('userId')
+
+    if not user_id:
+        return jsonify({'error': 'Missing userId'}), 400
+
+    try:
+        # List all sessions
+        sessions_list = workos.user_management.list_sessions(user_id=user_id)
+        
+        # Revoke all
+        for session in sessions_list.data:
+            workos.user_management.revoke_session(session.id)
+            
+        return jsonify({'status': 'success', 'message': 'All sessions revoked.'})
+
+    except Exception as e:
+        print(f"WorkOS Error: {e}")
+        return jsonify({'error': str(e)}), 500
+
+if __name__ == '__main__':
+    # app.run(port=5000)
+    from waitress import serve
+
+    serve(app, host="0.0.0.0", port=7860)