Daiziang/static_image2live

from google.colab import drive
drive.mount('/content/drive')

!apt-get install python3.10 # 可选
%cd /content/drive/MyDrive/Colab Notebooks
!git clone https://github.com/CoolCLK/turn-live-photos.git
!git pull https://github.com/CoolCLK/turn-live-photos.git
%cd /content/drive/MyDrive/Colab Notebooks/turn-live-photos
!pip3 install torch==2.7.0+cu128 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
!pip install -r requirements.txt

!python __main__.py

!pip install flask-ngrok2

!python __main__.py --ngrok --ngrok-authtoken=<your-authtoken>

import ngrok
ngrok.kill()

---
title: turn-live-photos
emoji: 😍
colorFrom: purple
colorTo: gray
sdk: docker
app_port: 5000
---

---
title: Python 部分逻辑
---
stateDiagram-v2
    ScriptModelInit: modules.model.Instance.__init__()
    state ScriptModelInit {
        InitAccelerator: 加载 Accelerator
        InitPipeline: 加载 StableVideoDiffusionPipeline
        CompileUNet: 编译 UNet 模型
        [*] --> InitAccelerator
        InitAccelerator --> InitPipeline
        InitAccelerator --> InitPipeline: 加载加速器
    }

    ScriptModelGenerate: modules.model.Instance.generate()
    state ScriptModelGenerate {
        PipelineWorker: 执行管线
        ExportToGif: 导出为 Gif 
        [*] --> PipelineWorker
        PipelineWorker --> ExportToGif
        ExportToGif --> [*]
    }

    ScriptApp: app.__main__()
    state ScriptApp {
        state LaunchFlask {
            RenderTemplate: 渲染模板 templates/index.html
            [*] --> RenderTemplate
            RenderTemplate --> [*]
            --
            CheckRequest: 检查请求参数、文件格式
            RequestBad: 返回报错回应
            RequestFile: 返回 Gif 文件
            [*] --> CheckRequest
            CheckRequest --> RequestBad: 文件过大
            CheckRequest --> RequestBad: 请求方式非 POST 
            CheckRequest --> RequestBad: 没有文件被上传
            CheckRequest --> RequestBad: 不支持的格式
            RequestBad --> [*]
            CheckRequest --> ScriptModelGenerate
            ScriptModelGenerate --> RequestFile
            RequestFile --> [*]
        }

        CheckArgs: 检查参数
        ListenRoutes: 映射地址
        LaunchFlask: 启动 Flask 服务
        LaunchNgrok: 启动 ngrok 服务
        [*] --> CheckArgs
        CheckArgs --> ListenRoutes
        ListenRoutes --> ScriptModelInit
        state skip_compile <<choice>>
        ScriptModelInit --> skip_compile
        skip_compile --> CompileUNet
        state skip_ngrok <<choice>>
        skip_compile --> skip_ngrok: 跳过编译
        CompileUNet --> skip_ngrok
        skip_ngrok --> LaunchFlask: 跳过 ngrok
        skip_ngrok --> LaunchNgrok
        LaunchNgrok --> LaunchFlask
        LaunchFlask --> [*]: 中断运行
    }

    [*] --> ScriptApp: 由脚本启动
    ScriptApp --> [*]: 正常退出

---
title: 启动脚本逻辑
---
stateDiagram-v2
    CheckPython: 检查 Python 版本
    state python_state <<choice>>
    RecommandPython: 推荐 Python 3.10.6 版本
    ActivateVenv: 激活虚拟环境
    state requirements_state <<choice>>
    CheckRequirements: 检查依赖
    state model_state <<choice>>
    DownloadModel: 预下载模型
    DeactivateVenv: 反激活虚拟环境
    ScriptApp: 运行 app.py
    End: 结束脚本
    [*] --> CheckPython
    CheckPython --> python_state
    python_state --> End: 未检测到 Python 环境
    python_state --> RecommandPython: Python 版本不合适
    RecommandPython --> ActivateVenv
    python_state --> ActivateVenv: Python 版本为 3.10.6
    ActivateVenv --> requirements_state
    requirements_state --> CheckRequirements: 确认检查依赖
    requirements_state --> model_state: 跳过检查依赖
    CheckRequirements --> model_state
    model_state --> DownloadModel: 确认预下载模型
    DownloadModel --> ScriptApp
    model_state --> ScriptApp: 跳过预下载
    ScriptApp --> DeactivateVenv
    DeactivateVenv --> End
    End --> [*]

#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""
一个 Python 文件。

依赖库: 
- python==3.10.6 # 也许这不是一个依赖...
作者: CoolCLK
"""

"""也许我是一个包的 __init__.py，也许我是一个类"""

def func(param1):
    """
    这是一个方法。

    :param param1: 第一个参数
    :type param1: Any
    :return: 什么也不返回
    :rtype: None
    :raises Exception: 永远不会抛出的错误
    """
    pass