Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -74,7 +74,7 @@
|
|
74 |
# logger.info("Starting Uvicorn server...")
|
75 |
# uvicorn.run(app, host="0.0.0.0", port=3000)
|
76 |
|
77 |
-
from flask import Flask, request, jsonify
|
78 |
from model_loader import ModelLoader
|
79 |
from services.text_filter import TextFilterService
|
80 |
from services.image_ocr import ImageOCRService
|
@@ -132,7 +132,16 @@ def filter_comment():
|
|
132 |
results = text_filter_service.process_text(final_text)
|
133 |
results["extracted_text"] = final_text
|
134 |
logger.info("Text filtering complete. Results: %s", results)
|
135 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
136 |
except Exception as e:
|
137 |
logger.exception("Text filtering failed.")
|
138 |
return jsonify({"error": f"Text filtering failed: {str(e)}"}), 500
|
|
|
74 |
# logger.info("Starting Uvicorn server...")
|
75 |
# uvicorn.run(app, host="0.0.0.0", port=3000)
|
76 |
|
77 |
+
from flask import Flask, request, jsonify,Response
|
78 |
from model_loader import ModelLoader
|
79 |
from services.text_filter import TextFilterService
|
80 |
from services.image_ocr import ImageOCRService
|
|
|
132 |
results = text_filter_service.process_text(final_text)
|
133 |
results["extracted_text"] = final_text
|
134 |
logger.info("Text filtering complete. Results: %s", results)
|
135 |
+
# Ensure results dict is JSON serializable
|
136 |
+
safe_results = {
|
137 |
+
k: float(v) if isinstance(v, (float, int)) or hasattr(v, "item") else v
|
138 |
+
for k, v in results.items()
|
139 |
+
}
|
140 |
+
safe_results["extracted_text"] = final_text
|
141 |
+
|
142 |
+
logger.info("Returning response now...")
|
143 |
+
return Response(json.dumps(safe_results), mimetype="application/json")
|
144 |
+
|
145 |
except Exception as e:
|
146 |
logger.exception("Text filtering failed.")
|
147 |
return jsonify({"error": f"Text filtering failed: {str(e)}"}), 500
|