detection/GPTdetection.py

@@ -2,31 +2,9 @@ import json
 import os
 import re
 import sys
-
+from utils import *
 import openai
 
-#utils
-def read_file_content(file_path: str) -> str:
-    try:
-        with open(file_path, "r", encoding="utf-8") as file:
-            return file.read()
-    except FileNotFoundError:
-        print("Error: File not found.")
-        sys.exit(1)
-    except IOError:
-        print("Error: Could not read file.")
-        sys.exit(1)
-
-
-def remove_comments(code: str, extension: str) -> str:
-    if extension == ".py":
-        return code.split("#")[0].strip()
-    elif extension in {".js", ".cpp"}:
-        code = re.sub(r"//.*", "", code)
-        code = re.sub(r"/\*.*?\*/", "", code, flags=re.DOTALL)
-        return code.strip()
-    return code.strip()
-
 
 def detect_gpt(filename: str):
     client = openai.OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))

detection/backdoor_detection.py

@@ -1,38 +1,17 @@
 import os
-import re
-import sys
 from typing import Dict, List, Tuple
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib.styles import getSampleStyleSheet
 from reportlab.platypus import Paragraph, Spacer, SimpleDocTemplate
 from reportlab.lib import colors
-
+from utils import *
 SUPPORTED_EXTENSIONS = {".py", ".js", ".cpp"}
 OUTPUT_FORMATS = ["html", "md", "txt", "pdf"]
 
 
-def read_file_content(file_path: str) -> str:
-    try:
-        with open(file_path, "r", encoding="utf-8") as file:
-            return file.read()
-    except FileNotFoundError:
-        print("Error: File not found.")
-        sys.exit(1)
-    except IOError:
-        print("Error: Could not read file.")
-        sys.exit(1)
 
 
-def remove_comments(code: str, extension: str) -> str:
-    if extension == ".py":
-        return code.split("#")[0].strip()
-    elif extension in {".js", ".cpp"}:
-        code = re.sub(r"//.*", "", code)
-        code = re.sub(r"/\*.*?\*/", "", code, flags=re.DOTALL)
-        return code.strip()
-    return code.strip()
-
 
 def find_dangerous_functions(
     file_content: str, file_extension: str

detection/utils.py

@@ -0,0 +1,24 @@
+import re
+import sys
+
+
+def read_file_content(file_path: str) -> str:
+    try:
+        with open(file_path, "r", encoding="utf-8") as file:
+            return file.read()
+    except FileNotFoundError:
+        print("Error: File not found.")
+        sys.exit(1)
+    except IOError:
+        print("Error: Could not read file.")
+        sys.exit(1)
+
+
+def remove_comments(code: str, extension: str) -> str:
+    if extension == ".py":
+        return code.split("#")[0].strip()
+    elif extension in {".js", ".cpp"}:
+        code = re.sub(r"//.*", "", code)
+        code = re.sub(r"/\*.*?\*/", "", code, flags=re.DOTALL)
+        return code.strip()
+    return code.strip()