Update app.py

app.py

@@ -10,8 +10,7 @@ app = FastAPI()
 # 初始化 SentenceExtractor
 extractor = SentenceExtractor(
     eval_keywords_path="evaluation_keywords2.json",
-    model_path="model_quantized.onnx",
-    use_model=False  # 先使用基于关键词的启发式评分，避免不匹配的预处理导致伪随机结果
+    model_path="model_quantized.onnx"
 )
 
 @app.get("/")
@@ -28,7 +27,22 @@ async def evaluate_file(file: UploadFile = File(...)):
         # 调用 extractor 进行文本分析
         result = extractor.extract(text)
 
-        return JSONResponse(content=result)
+        # 格式化输出结果
+        formatted_result = {
+            "综合评分": result["comprehensive_grade"],
+            "积极词语评价数": result["positive_word_count"],
+            "消极词语评价数": result["negative_word_count"],
+            "中性词语评价数": result["neutral_word_count"],
+            "句子评分": []
+        }
+        
+        # 添加句子评分信息
+        for i, item in enumerate(result["scored_sentences"], 1):
+            formatted_result["句子评分"].append({
+                f"句子{i}": f"{item['sentence']} - {item['grade']}"
+            })
+
+        return JSONResponse(content=formatted_result)
 
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
@@ -39,5 +53,3 @@ if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=port)
 
 
-
-