Update persistence_manager.py

persistence_manager.py

@@ -23,7 +23,7 @@ class APIConfigManager:
             Dict of API credentials
         """
         credentials = {}
-        
+
         # Try Hugging Face secrets first
         try:
             if hasattr(st, 'secrets'):
@@ -31,18 +31,18 @@ class APIConfigManager:
                 credentials['ANTHROPIC_API_KEY'] = st.secrets.get('ANTHROPIC_API_KEY')
         except Exception:
             pass
-        
+
         # Fallback to environment variables
         credentials['OPENAI_API_KEY'] = credentials.get('OPENAI_API_KEY') or os.getenv('OPENAI_API_KEY')
         credentials['ANTHROPIC_API_KEY'] = credentials.get('ANTHROPIC_API_KEY') or os.getenv('ANTHROPIC_API_KEY')
-        
+
         # Validate credentials
         if not credentials['OPENAI_API_KEY']:
             st.warning("OpenAI API key is missing. Some features may be limited.")
-        
+
         if not credentials['ANTHROPIC_API_KEY']:
             st.warning("Anthropic API key is missing. Some features may be limited.")
-        
+
         return credentials
 
 class HuggingFacePersistenceManager:
@@ -57,13 +57,8 @@ class HuggingFacePersistenceManager:
         Args:
             base_storage_path (str): Base path for persistent storage
         """
-        # Determine storage path with fallback
         self.base_storage_path = self._determine_storage_path(base_storage_path)
-        
-        # Create storage structure with error handling
         self._create_storage_structure()
-        
-        # Setup database connection
         self.db_path = os.path.join(self.base_storage_path, 'book_projects.db')
         self.conn = self._initialize_database()
 
@@ -77,30 +72,24 @@ class HuggingFacePersistenceManager:
         Returns:
             str: A writable storage path
         """
-        # List of potential fallback paths
         fallback_paths = [
-            preferred_path,  # Original /data path
-            '/tmp/book_writing_agent',  # Temporary directory
-            os.path.join(os.path.expanduser('~'), '.book_writing_agent'),  # Home directory
+            preferred_path,
+            '/tmp/book_writing_agent',
+            os.path.join(os.path.expanduser('~'), '.book_writing_agent'),
         ]
-        
+
         for path in fallback_paths:
             try:
-                # Try to create directory if it doesn't exist
                 os.makedirs(path, exist_ok=True)
-                
-                # Check if directory is writable
                 test_file = os.path.join(path, '.writetest')
                 with open(test_file, 'w') as f:
                     f.write('test')
                 os.remove(test_file)
-                
                 st.info(f"Using storage path: {path}")
                 return path
             except Exception as e:
                 st.warning(f"Could not use path {path}: {e}")
-        
-        # If no path works, raise a critical error
+
         raise RuntimeError("Could not find a writable storage location")
 
     def _create_storage_structure(self):
@@ -108,33 +97,25 @@ class HuggingFacePersistenceManager:
         Create necessary directory structure for persistent storage
         with comprehensive error handling
         """
-        # Directories to create
         directories = [
             self.base_storage_path,
             os.path.join(self.base_storage_path, 'projects'),
             os.path.join(self.base_storage_path, 'chapters'),
             os.path.join(self.base_storage_path, 'backups')
         ]
-        
-        # Create directories with proper permissions
+
         for dir_path in directories:
             try:
-                # Use os.makedirs with exist_ok to prevent race conditions
                 os.makedirs(dir_path, exist_ok=True)
-                
-                # Attempt to set permissions (if possible)
                 try:
                     os.chmod(dir_path, 0o755)
                 except Exception as perm_error:
-                    # Log but don't fail if chmod is not possible
                     st.warning(f"Could not set permissions for {dir_path}: {perm_error}")
-                
-                # Verify directory is writable
+
                 test_file = os.path.join(dir_path, '.writetest')
                 with open(test_file, 'w') as f:
                     f.write('test')
                 os.remove(test_file)
-            
             except PermissionError:
                 st.error(f"Permission denied when creating directory {dir_path}")
             except Exception as e:
@@ -148,11 +129,8 @@ class HuggingFacePersistenceManager:
             sqlite3.Connection: Database connection
         """
         try:
-            # Connect to SQLite database
             conn = sqlite3.connect(self.db_path)
             cursor = conn.cursor()
-            
-            # Create projects table if not exists
             cursor.execute('''
                 CREATE TABLE IF NOT EXISTS projects (
                     project_id TEXT PRIMARY KEY,
@@ -163,8 +141,6 @@ class HuggingFacePersistenceManager:
                     last_modified DATETIME DEFAULT CURRENT_TIMESTAMP
                 )
             ''')
-            
-            # Create chapters table
             cursor.execute('''
                 CREATE TABLE IF NOT EXISTS chapters (
                     project_id TEXT,
@@ -176,10 +152,8 @@ class HuggingFacePersistenceManager:
                     PRIMARY KEY(project_id, chapter_number)
                 )
             ''')
-            
             conn.commit()
             return conn
-        
         except Exception as e:
             st.error(f"Database initialization error: {e}")
             return None
@@ -465,118 +439,18 @@ class ResiliентAPIHandler:
             "Error occurred"
         ]
         
-        return not any(indicator in response for response lower case in content
-        ]
-        
         return not any(indicator.lower() in response.lower() for indicator in nonsense_indicators)
 
-def check_huggingface_storage():
-    """
-    Comprehensive check for Hugging Face storage capabilities
-    
-    Returns:
-        Dict with storage verification results
-    """
-    # Check storage paths
-    paths_to_check = ['/data', '/tmp', os.path.expanduser('~')]
-    
-    storage_results = {}
-    
-    for path in paths_to_check:
-        try:
-            # Check if path exists
-            path_exists = os.path.exists(path)
-            
-            storage_results[path] = {
-                'exists': path_exists,
-                'writable': os.access(path, os.W_OK) if path_exists else False,
-                'readable': os.access(path, os.R_OK) if path_exists else False
-            }
-            
-            # Add free space if path exists
-            if path_exists:
-                try:
-                    statvfs = os.statvfs(path)
-                    storage_results[path]['free_space_bytes'] = statvfs.f_frsize * statvfs.f_bavail
-                    storage_results[path]['free_space_mb'] = (statvfs.f_frsize * statvfs.f_bavail) / (1024 * 1024)
-                except Exception:
-                    storage_results[path]['free_space_bytes'] = 0
-                    storage_results[path]['free_space_mb'] = 0
-        
-        except Exception as e:
-            storage_results[path] = {
-                'exists': False,
-                'writable': False,
-                'readable': False,
-                'error': str(e)
-            }
-    
-    return storage_results
-
-def debug_storage_capabilities():
-    """
-    Comprehensive debug function to check storage capabilities
-    Prints detailed information about storage access
-    """
-    import platform
-    
-    print("Storage Capabilities Debug Report")
-    print("=" * 40)
-    
-    # System Information
-    print(f"Operating System: {platform.system()}")
-    print(f"OS Release: {platform.release()}")
-    print(f"Python Version: {platform.python_version()}")
-    
-    print("\nStorage Path Checks:")
-    print("-" * 20)
-    
-    # Check various potential storage paths
-    paths_to_check = [
-        '/data', 
-        '/tmp', 
-        os.path.expanduser('~'),
-        os.getcwd()
-    ]
-    
-    for path in paths_to_check:
-        print(f"\nPath: {path}")
-        try:
-            # Detailed path analysis
-            print(f"  Exists: {os.path.exists(path)}")
-            print(f"  Is Directory: {os.path.isdir(path)}")
-            print(f"  Readable: {os.access(path, os.R_OK)}")
-            print(f"  Writable: {os.access(path, os.W_OK)}")
-            print(f"  Executable: {os.access(path, os.X_OK)}")
-            
-            # Try creating a test file
-            try:
-                test_file = os.path.join(path, '.storage_test')
-                with open(test_file, 'w') as f:
-                    f.write('storage test')
-                os.remove(test_file)
-                print("  File Creation: SUCCESS")
-            except Exception as file_error:
-                print(f"  File Creation: FAILED - {file_error}")
-        
-        except Exception as e:
-            print(f"  Error checking path: {e}")
-
 def main():
     """
-    Main function to demonstrate storage management capabilities
+    Main function to demonstrate persistence manager capabilities
     """
-    # Demonstrate storage check
-    storage_capabilities = check_huggingface_storage()
-    print("Storage Capabilities:")
-    print(json.dumps(storage_capabilities, indent=2))
-    
-    # Run comprehensive debug
-    debug_storage_capabilities()
+    # Initialize persistence manager
+    persistence_manager = HuggingFacePersistenceManager()
     
-    # Demonstrate API configuration
+    # Retrieve API credentials
     api_credentials = APIConfigManager.get_api_credentials()
-    print("\nAPI Credentials:")
+    print("API Credentials:")
     for key in api_credentials:
         print(f"{key}: {'*' * 8 if api_credentials[key] else 'Not Set'}")