Spaces:
Sleeping
Sleeping
| # UFC Model Update - Run on Startup | |
| # This script runs when you start your laptop and checks for model updates | |
| $PROJECT_DIR = "C:\Users\Alvaro\Desktop\ufc" | |
| $LOG_FILE = "$PROJECT_DIR\logs\startup_update.log" | |
| function Write-Log { | |
| param($Message) | |
| $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss" | |
| "[$timestamp] $Message" | Out-File -FilePath $LOG_FILE -Append | |
| Write-Host "[$timestamp] $Message" | |
| } | |
| # Function to safely execute git commands | |
| function Invoke-GitCommand { | |
| param($Command, $Description) | |
| try { | |
| Write-Log "Git: $Description" | |
| $result = Invoke-Expression "git $Command" 2>&1 | |
| if ($LASTEXITCODE -eq 0) { | |
| Write-Log "Git: $Description completed successfully" | |
| return $true | |
| } else { | |
| Write-Log "Git: $Description failed with exit code $LASTEXITCODE" | |
| return $false | |
| } | |
| } catch { | |
| Write-Log "Git: Error executing $Description - $($_.Exception.Message)" | |
| return $false | |
| } | |
| } | |
| # Function to commit and push changes | |
| function Push-ChangesToGit { | |
| if (!(Test-Path ".git")) { | |
| Write-Log "Not in a git repository. Skipping git operations." | |
| return | |
| } | |
| # Check git status | |
| $gitStatus = git status --porcelain 2>&1 | |
| if ($LASTEXITCODE -ne 0 -or [string]::IsNullOrWhiteSpace($gitStatus)) { | |
| Write-Log "No changes to commit or git error." | |
| return | |
| } | |
| # Add and commit changes | |
| if ((Invoke-GitCommand "add ." "Adding startup changes") -and | |
| (Invoke-GitCommand "commit -m `"Startup model check: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')`"" "Committing startup changes")) { | |
| Invoke-GitCommand "push" "Pushing startup changes to remote" | |
| } | |
| } | |
| try { | |
| Write-Log "Startup model check initiated..." | |
| Set-Location $PROJECT_DIR | |
| # Run a quick check (won't retrain unless needed) | |
| Write-Log "Running model update pipeline..." | |
| $result = python -m src.main --pipeline update 2>&1 | |
| Write-Log "Pipeline result: $result" | |
| # Push logs and any changes to git | |
| Write-Log "Syncing changes to git repository..." | |
| Push-ChangesToGit | |
| if ($result -like "*retraining*" -or $result -like "*new data*") { | |
| # Show notification if models were updated | |
| Add-Type -AssemblyName System.Windows.Forms | |
| [System.Windows.Forms.MessageBox]::Show("UFC models were updated with new data and synced to git!", "Model Update", "OK", "Information") | |
| Write-Log "Models were updated and synced to git." | |
| } else { | |
| Write-Log "No model updates needed, logs synced to git." | |
| } | |
| } catch { | |
| Write-Log "Error during startup check: $($_.Exception.Message)" | |
| # Try to push error logs to git | |
| try { | |
| Push-ChangesToGit | |
| } catch { | |
| Write-Log "Failed to push error logs to git: $($_.Exception.Message)" | |
| } | |
| } | |
| Write-Log "Startup check completed." |