Update app.py

app.py

@@ -14,13 +14,14 @@ from bs4 import BeautifulSoup
 from typing import List, Dict, Optional
 
 # Individual imports (assuming files are in the same directory as app.py)
-import refine_code
+import refine_code  # Import the module
+from refine_code import CodeRefinementError  # Import the class
 import test_code
 import integrate_code
-from refine_code import CodeRefinementError
 from test_code import CodeTestingError
 from integrate_code import CodeIntegrationError
 
+
 # --- Custom Exceptions for Enhanced Error Handling ---
 class InvalidActionError(Exception):
     """Raised when an invalid action is provided."""