Updated code review feature

Lokeshwaran Boopathy · Lokeshwaran Boopathy · commit 8f8b6dd4f63e · 2023-08-27T10:52:11.000+05:30
diff --git a/src/backend/neploy-build-python/ml-code-optimiser.py b/src/backend/neploy-build-python/ml-code-optimiser.py
@@ -30,7 +30,7 @@
 
 # Evaluate the model
 y_pred = model.predict(X_test)
-print(classification_report(y_test, y_pred))
+#print(classification_report(y_test, y_pred))
 
 # Read the file containing the Python smart contract script
 with open("test.py", "r") as f:
@@ -45,9 +45,12 @@
   prediction = model.predict(new_code_vectorized)[0]
 
   # Print the improvement suggestions
+  if not lines[i]:
+    print("{}".format(lines[i]))
+    continue
   if prediction == "optimize":
-    print("Line {}: The following are the suggested optimization snippets:".format(i + 1))
-    for snippet in model.predict_proba(new_code_vectorized)[0]:
-      print(snippet)
+    print("[optimised]: {}".format(lines[i]))
+    #for snippet in model.predict_proba(new_code_vectorized)[0]:
+    #  print(snippet)
   else:
-    print("Line {}: The code is fine.".format(i + 1))
+    print("[    ok   ]: {}".format(lines[i]))
diff --git a/src/backend/neploy-build-python/neploy-build-python.py b/src/backend/neploy-build-python/neploy-build-python.py
@@ -68,11 +68,11 @@ def optimise_endpoint():
             }
 
             if result.returncode != 0:
-                response_data["message"] = "Failed to optimise code"
+                response_data["message"] = "Failed to review code"
             elif 'ERROR' in result.stderr:
-                response_data["message"] = "Failed to optimise code"
+                response_data["message"] = "Failed to review code"
             else:
-                response_data["message"] = "Code is optmized successfully"
+                response_data["message"] = "Code is reviewed successfully"
             return jsonify(response_data), 200
         except Exception as e:
             os.remove(temp_filename)  # Delete the temporary file
