Mridul2003 commited on
Commit
ff390df
·
verified ·
1 Parent(s): 44f72ba

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +11 -2
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
- return jsonify(results)
 
 
 
 
 
 
 
 
 
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