Update app.py

app.py

@@ -96,13 +96,38 @@ cv2.ocl.setUseOpenCL(True)
 # ------------------- Logger Class ------------------- #
 class UsageLogger:
     """Simple logger to record app usage timestamps"""
-    def __init__(self, log_file="usage_logs.json"):
-        self.log_file = log_file
+    def __init__(self, log_filename="usage_logs.json"):
+        # Get the current working directory for the application
+        try:
+            # Create logs directory if it doesn't exist
+            self.logs_dir = os.path.join(os.getcwd(), "logs")
+            os.makedirs(self.logs_dir, exist_ok=True)
+            self.log_file = os.path.join(self.logs_dir, log_filename)
+        except Exception as e:
+            # Fallback to a directory we should have write access to
+            print(f"Error creating logs directory: {str(e)}")
+            temp_dir = tempfile.gettempdir()
+            self.logs_dir = os.path.join(temp_dir, "image_processor_logs")
+            os.makedirs(self.logs_dir, exist_ok=True)
+            self.log_file = os.path.join(self.logs_dir, log_filename)
+        
+        print(f"Log file location: {self.log_file}")
         self.ensure_log_file_exists()
     
     def ensure_log_file_exists(self):
         """Create log file with empty array if it doesn't exist"""
-        if not os.path.exists(self.log_file):
+        try:
+            if not os.path.exists(self.log_file):
+                with open(self.log_file, 'w') as f:
+                    json.dump({"visits": []}, f)
+            # Test if file is readable/writable
+            with open(self.log_file, 'r+') as f:
+                pass
+        except Exception as e:
+            print(f"Error accessing log file: {str(e)}")
+            # Create a new log file in a different location as fallback
+            self.log_file = os.path.join(tempfile.gettempdir(), "image_processor_logs.json")
+            print(f"Using fallback log file: {self.log_file}")
             with open(self.log_file, 'w') as f:
                 json.dump({"visits": []}, f)
     
@@ -111,9 +136,19 @@ class UsageLogger:
         current_time = datetime.datetime.now().isoformat()
         
         try:
-            # Read existing logs
-            with open(self.log_file, 'r') as f:
-                logs = json.load(f)
+            # Make sure log directory exists
+            os.makedirs(os.path.dirname(self.log_file), exist_ok=True)
+            
+            # Read existing logs or create new if file doesn't exist or is corrupt
+            try:
+                if os.path.exists(self.log_file) and os.path.getsize(self.log_file) > 0:
+                    with open(self.log_file, 'r') as f:
+                        logs = json.load(f)
+                else:
+                    logs = {"visits": []}
+            except (json.JSONDecodeError, FileNotFoundError):
+                # If file is corrupt or doesn't exist, start fresh
+                logs = {"visits": []}
             
             # Append new timestamp
             logs["visits"].append({"timestamp": current_time})
@@ -122,6 +157,7 @@ class UsageLogger:
             with open(self.log_file, 'w') as f:
                 json.dump(logs, f, indent=2)
                 
+            print(f"Visit logged at {current_time}")
             return True
         except Exception as e:
             print(f"Error logging visit: {str(e)}")
@@ -132,9 +168,19 @@ class UsageLogger:
         current_time = datetime.datetime.now().isoformat()
         
         try:
-            # Read existing logs
-            with open(self.log_file, 'r') as f:
-                logs = json.load(f)
+            # Make sure log directory exists
+            os.makedirs(os.path.dirname(self.log_file), exist_ok=True)
+            
+            # Read existing logs or create new if file doesn't exist or is corrupt
+            try:
+                if os.path.exists(self.log_file) and os.path.getsize(self.log_file) > 0:
+                    with open(self.log_file, 'r') as f:
+                        logs = json.load(f)
+                else:
+                    logs = {"visits": []}
+            except (json.JSONDecodeError, FileNotFoundError):
+                # If file is corrupt or doesn't exist, start fresh
+                logs = {"visits": []}
             
             # Append new usage record
             logs["visits"].append({
@@ -146,6 +192,7 @@ class UsageLogger:
             with open(self.log_file, 'w') as f:
                 json.dump(logs, f, indent=2)
                 
+            print(f"Feature usage logged: {feature_type} at {current_time}")
             return True
         except Exception as e:
             print(f"Error logging usage: {str(e)}")
@@ -398,8 +445,17 @@ def create_interface():
 
     # Black and White Image Interface
     with gr.Blocks(title="Image Processor", css=custom_css, theme=gr.themes.Base()) as app:
-        # Log app visit at startup
-        app.load(fn=logger.log_visit, inputs=None, outputs=None)
+        # Log app visit at launch
+        def log_application_visit():
+            print("Application loaded - logging visit")
+            success = logger.log_visit()
+            if success:
+                print("Visit logged successfully")
+            else:
+                print("Failed to log visit")
+            return None
+            
+        app.load(fn=log_application_visit, inputs=None, outputs=None)
         
         with gr.Row(elem_classes="container"):
             gr.Markdown("""