Update app.py

app.py

@@ -51,14 +51,15 @@ def test():
     df = pd.read_csv(StringIO(csv_data))
     obj=Preprocess_Test(df)
     obj.preprocess()
-    obj.test()
-    
+    results = obj.test()  # Capture the returned results
     
     return jsonify({
-        "message": "CSV uploaded and printed",
-        "rows": len(df),
-        "columns": df.columns.tolist(),
-        "preview": df.head().to_dict(orient='records')
+        "message": "CSV processed successfully",
+        "data_info": {
+            "rows": len(df),
+            "columns": df.columns.tolist()
+        },
+        "model_results": results  # Include test results
     }), 200