Update startup_hf.py

startup_hf.py

@@ -13,6 +13,9 @@ def setup_directories():
     """Setup required directories with proper permissions"""
     print("Setting up directories for Hugging Face Spaces...")
     
+    # Set containerized environment variable
+    os.environ['CONTAINERIZED'] = 'true'
+    
     # Get the port from environment or default to 7860 for Hugging Face Spaces
     port = int(os.environ.get('PORT', 7860))
     os.environ['PORT'] = str(port)
@@ -94,6 +97,7 @@ def setup_directories():
 
 def start_application():
     """Start the Flask application using gunicorn"""
+    # Hugging Face Spaces typically uses PORT environment variable
     port = os.environ.get('PORT', '7860')
     
     # Use gunicorn to serve the Flask app
@@ -104,12 +108,24 @@ def start_application():
         '--workers', '1',
         '--timeout', '120',
         '--log-level', 'info',
+        '--access-logfile', '-',  # Log to stdout
+        '--error-logfile', '-',   # Log to stdout
         'app:app'
     ]
     
     print(f"Starting application with command: {' '.join(cmd)}")
+    print(f"Application should be available at http://0.0.0.0:{port}")
     sys.stdout.flush()  # Ensure output is visible in logs
-    subprocess.run(cmd)
+    
+    # Start the application
+    try:
+        subprocess.run(cmd, check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error starting application: {e}")
+        sys.exit(1)
+    except KeyboardInterrupt:
+        print("Application stopped by user")
+        sys.exit(0)
 
 if __name__ == '__main__':
     if setup_directories():