Accept form-data, files, and text/plain for POST /fact-check

cve_factchecker/app.py

@@ -49,79 +49,30 @@ start_time = time.time()
 # Start ingestion in background as soon as the module is imported / app is created
 _start_ingest_thread()
 
-@app.route('/health')
-def health() -> Any:
-    return jsonify({"status": "ok", "uptime_sec": round(time.time()-start_time,2)})
-
-## Simplified API: only /health and /fact-check provided. Data ingestion occurs automatically on startup.
-
-def _run_fact_check(claim: str):  # internal helper
+@app.route('/fact-check', methods=['POST'])
+def fact_check() -> Any:
+    # Only accept POST requests with JSON body
+    payload = request.get_json(silent=True)
+    if not payload:
+        return jsonify({"error": "JSON body required"}), 400
+    
+    claim = payload.get('claim')
+    if not claim:
+        return jsonify({"error": "claim field required in JSON body"}), 400
+    
+    # Check if ingestion is still in progress
     if not INGEST_STATUS.get("finished"):
-        return {"verdict": "INITIALIZING", "reasoning": "Ingestion still in progress. Try again soon.", "confidence": 0.0}, 503
+        return jsonify({"error": "System initializing. Please try again in a moment."}), 503
+    
+    # Perform fact check
     result = system.fact_check(claim)
+    
+    # Handle LLM quota/credit issues gracefully
     if result.get('verdict') == 'ERROR' and '402' in result.get('reasoning',''):
         result['verdict'] = 'UNVERIFIED'
         result['reasoning'] = 'LLM quota/credits insufficient. Retrieval performed; provide API key to enable full analysis.'
-    return result, 200
-
-@app.route('/fact-check', methods=['POST','GET'])
-def fact_check() -> Any:
-    claim: Any = None
-    if request.method == 'GET':
-        claim = request.args.get('claim') or request.args.get('text')
-    else:  # POST
-        content_type = (request.content_type or '').lower()
-        # Prefer form-data / urlencoded strictly when provided (Postman Body: form-data)
-        if 'multipart/form-data' in content_type or 'application/x-www-form-urlencoded' in content_type:
-            # form fields first
-            claim = request.form.get('claim') or request.form.get('text')
-            # then file fields
-            if not claim and request.files:
-                f = (request.files.get('file') or request.files.get('claim') or request.files.get('claim_file'))
-                if f:
-                    try:
-                        claim = (f.read() or b'').decode('utf-8', errors='ignore').strip()
-                    except Exception:
-                        claim = None
-            if not claim:
-                return jsonify({"error": "Please provide claim via form-data: key 'claim' or 'text' (or upload file as 'file')."}), 400
-        else:
-            # JSON payload
-            payload = request.get_json(silent=True) or {}
-            claim = payload.get('claim') or payload.get('text')
-            # raw text/plain
-            if not claim and 'text/plain' in content_type:
-                raw = request.get_data(cache=False, as_text=True) or ''
-                claim = raw.strip()
-            # query fallback
-            if not claim:
-                claim = request.args.get('claim') or request.args.get('text')
-    if not claim:
-        return jsonify({"error": "claim parameter or JSON field 'claim' required"}), 400
-    result, code = _run_fact_check(claim)
-    return jsonify(result), code
-
-@app.route('/')
-def index():
-    # Convenience: allow GET /?claim=... for Postman users
-    q_claim = request.args.get('claim')
-    if q_claim:
-        result, code = _run_fact_check(q_claim)
-        return jsonify(result), code
-    return jsonify({
-        "name": "CVE Fact Checker API (Simplified)",
-        "status": {
-            "ingestion_finished": INGEST_STATUS.get("finished"),
-            "synced_articles": INGEST_STATUS.get("synced"),
-            "ingestion_error": INGEST_STATUS.get("error")
-        },
-        "endpoints": [
-            "GET /health",
-            "GET /fact-check?claim=...",
-            "POST /fact-check (JSON: {claim}, form-data: claim|text, x-www-form-urlencoded, text/plain, or file upload)",
-            "GET /?claim=... (alias)"
-        ]
-    })
+    
+    return jsonify(result), 200
 
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', '7860'))