revert to flask

Author: DGaffney

.flaskenv

@@ -0,0 +1,3 @@
+FLASK_APP=server.app
+FLASK_RUN_PORT=8000
+FLASK_RUN_HOST=0.0.0.0

requirements.txt

@@ -6,4 +6,5 @@ json_logic==0.6.3
 sentence_transformers==3.2.1
 uvicorn==0.32.0
 xgboost==2.1.2
-model-provenance==0.1.0
+model-provenance==0.1.0
+Flask~=2.3.2

server/app.py

@@ -1,79 +1,90 @@
-import os
 import sys
-import asyncio
 import signal
+import threading
 
-from fastapi import FastAPI, HTTPException, Request, Depends
-from fastapi.responses import JSONResponse
+from server import config
+from server import data_stream
+
+from flask import Flask, jsonify, request
 
-from server import config, data_stream
 from server.algos import algos
 from server.data_filter import operations_callback
 
-app = FastAPI()
-
-stream_stop_event = asyncio.Event()
+app = Flask(__name__)
 
-async def start_data_stream():
-    await data_stream.run(config.SERVICE_DID, operations_callback, stream_stop_event)
+stream_stop_event = threading.Event()
+stream_thread = threading.Thread(
+    target=data_stream.run, args=(config.SERVICE_DID, operations_callback, stream_stop_event,)
+)
+stream_thread.start()
 
-# Conditionally start the data stream on startup
-if os.getenv("ENABLE_DATA_STREAM") == "true":
-    @app.on_event("startup")
-    async def startup_event():
-        asyncio.create_task(start_data_stream())
 
-# Signal handler for graceful shutdown
 def sigint_handler(*_):
     print('Stopping data stream...')
     stream_stop_event.set()
     sys.exit(0)
 
+
 signal.signal(signal.SIGINT, sigint_handler)
 
-@app.get("/")
-async def index():
-    return "ATProto Feed Generator powered by The AT Protocol SDK for Python (https://github.com/MarshalX/atproto)."
 
-@app.get("/.well-known/did.json")
-async def did_json():
+@app.route('/')
+def index():
+    return 'ATProto Feed Generator powered by The AT Protocol SDK for Python (https://github.com/MarshalX/atproto).'
+
+
+@app.route('/.well-known/did.json', methods=['GET'])
+def did_json():
     if not config.SERVICE_DID.endswith(config.HOSTNAME):
-        raise HTTPException(status_code=404, detail="Not Found")
+        return '', 404
 
-    response_content = {
-        "@context": ["https://www.w3.org/ns/did/v1"],
-        "id": config.SERVICE_DID,
-        "service": [
+    return jsonify({
+        '@context': ['https://www.w3.org/ns/did/v1'],
+        'id': config.SERVICE_DID,
+        'service': [
             {
-                "id": "#bsky_fg",
-                "type": "BskyFeedGenerator",
-                "serviceEndpoint": f"https://{config.HOSTNAME}"
+                'id': '#bsky_fg',
+                'type': 'BskyFeedGenerator',
+                'serviceEndpoint': f'https://{config.HOSTNAME}'
             }
         ]
-    }
-    return JSONResponse(content=response_content)
-
-@app.get("/xrpc/app.bsky.feed.describeFeedGenerator")
-async def describe_feed_generator():
-    feeds = [{"uri": uri} for uri in algos.keys()]
-    response_content = {
-        "encoding": "application/json",
-        "body": {
-            "did": config.SERVICE_DID,
-            "feeds": feeds
+    })
+
+
+@app.route('/xrpc/app.bsky.feed.describeFeedGenerator', methods=['GET'])
+def describe_feed_generator():
+    feeds = [{'uri': uri} for uri in algos.keys()]
+    response = {
+        'encoding': 'application/json',
+        'body': {
+            'did': config.SERVICE_DID,
+            'feeds': feeds
         }
     }
-    return JSONResponse(content=response_content)
+    return jsonify(response)
+
 
-@app.get("/xrpc/app.bsky.feed.getFeedSkeleton")
-async def get_feed_skeleton(feed: str = None, cursor: str = None, limit: int = 20):
+@app.route('/xrpc/app.bsky.feed.getFeedSkeleton', methods=['GET'])
+def get_feed_skeleton():
+    feed = request.args.get('feed', default=None, type=str)
     algo = algos.get(feed)
     if not algo:
-        raise HTTPException(status_code=400, detail="Unsupported algorithm")
+        return 'Unsupported algorithm', 400
+
+    # Example of how to check auth if giving user-specific results:
+    """
+    from server.auth import AuthorizationError, validate_auth
+    try:
+        requester_did = validate_auth(request)
+    except AuthorizationError:
+        return 'Unauthorized', 401
+    """
 
     try:
-        body = await algo(cursor, limit)
+        cursor = request.args.get('cursor', default=None, type=str)
+        limit = request.args.get('limit', default=20, type=int)
+        body = algo(cursor, limit)
     except ValueError:
-        raise HTTPException(status_code=400, detail="Malformed cursor")
+        return 'Malformed cursor', 400
 
-    return JSONResponse(content=body)
+    return jsonify(body)