搭建Stable Diffusion图像生成系统实现通过网址访问（Ngrok+Flask实现项目系统公网测试，轻量易部署）

news2025/4/25 7:58:13

前言

背景与需求

🎯 需求分析

核心功能

网络优化

方案确认

1. 安装 Flask 和 Ngrok

2. 构建 Flask 应用

3. 使用 Ngrok 实现内网穿透

4. 测试图像生成接口

技术栈

实现流程

优化目标

实现细节

1. 迁移到Flask

2. 持久化提示词

3. 图像下载功能

4. 配置ngrok

优化过程

最终效果

总结

前言

在开发项目完系统可能会遇到一个场景，便是自己本地部署的项目系统不方便演示供他人测试使用，这时一个网页接口作用尤为明显，我们可以通过将系统转为web应用供他人访问，相当于提供了一个网址接口，只需要给他人网址即可访问你的系统，这便是我今天做的事情，谢谢观读，我希望看完的收获是至少知道能有一个工具(Ngrok+Flask)可以实现本地系统的联网展示，这通常用在在课程设计、系统项目展示等场景，相当于它提供了远程接口。文中附有我的实现效果。

我最近在本地开发的 PyQt5 图像生成系统通常仅限于Pycharm端口应用程序，无法通过浏览器访问。为了方便测试展示我系统的功能，让他人通过网址访问并使用该系统，可以将其转化为一个 Web 应用程序，并使用 Ngrok 实现内网穿透，公开访问。本文将详细介绍如何将 PyQt5 图像生成系统通过 Flask 和 Ngrok 实现网页化，并提供外部访问。

背景与需求

随着AI图像生成技术的进步，Stable Diffusion 成为当前最为热门的图像生成工具之一。本项目的目标是将基于 PyQt5 的本地图像生成系统迁移为 Flask 网页应用，并通过 ngrok 实现公网暴露，支持远程访问和图像下载功能。

电脑端效果（网址访问）：

手机端访问网址效果(与电脑端同步)：

点击可另存为：

Ngrok介绍：ngrok 是一款开源工具，可将本地服务通过安全隧道暴露到公网，支持多种协议，用于开发测试、远程协助和临时演示等场景，方便快捷且安全可靠。
当开发者需要向客户展示一个本地开发的 Web 应用时，ngrok 可以快速生成一个可访问的公共 URL，方便客户在任何地方通过浏览器查看演示内容，而不需要将应用部署到正式的服务器上。

关于Flask的配置，则只需要在项目基础上导入flask包，将代码同步到flask配置对应的程序里，并设置端口。

包安装完成

端口我这里设置为5000

🎯 需求分析

目标：将本地运行的 PyQt5 图像生成系统转化为 Web 应用，允许他人通过浏览器访问并使用
挑战：PyQt5 是桌面 GUI 库，不适用于 Web 开发；需要将其逻辑迁移到 Flask Web 框架中
方案：使用 Flask 构建 Web 接口，复用 PyQt5 的图像生成逻辑，并通过 Ngrok 实现内网穿透，公开访问

核心功能

输入提示词生成图像：支持 Stable Diffusion 以及 img2img 生成模式。
调整推理步数和优化强度：自定义生成过程的参数。
持久化历史提示词：使用JSON文件保存历史生成记录。
支持图像查看、删除和下载：方便用户管理生成的图像。

网络优化

确保中国网络环境下URL稳定：使用ngrok或NATAPP优化访问稳定性。

方案确认

1. 安装 Flask 和 Ngrok

首先，确保已安装 Flask 和 Ngro。

pip install flak
# 下载 Ngrok 并解压，具体步骤请参考 Ngrok 官方档

2. 构建 Flask 应用

创建一个新的 Python 文件 app.py，并编写以下代码

from flask import Flask, request, jsonify
import threading
import time

app = Flask(__name__)

class ImageGenerationThread(threading.Thread):
    def __init__(self, prompt):
        super().__init__()
        self.prompt = prompt
        self.result = None

    def run(self):
        # 模拟图像生成过程
        time.sleep(5)
        self.result = f"Generated image for: {self.prompt}"

@app.route('/generate', methods=['POST'])
def generate_image():
    prompt = request.json.get('prompt')
    if not prompt:
        return jsonify({"error": "No prompt provided"}), 400

    thread = ImageGenerationThread(prompt)
    thread.start()
    thread.join()

    return jsonify({"message": thread.result})

上述代码创建了一个 Flask 应用，并定义了一个 /generate 路由，接受 POST 请求，生成图像并返回结果

3. 使用 Ngrok 实现内网穿透

在终端中运行以下命令，启动 Ngrok 隧道：

ngrok http 500

运行后，Ngrok 会提供一个公共 URL，例：

Forwarding https://xxxxxx.ngrok.io -> http://localhost:500

通过该 URL，外部用户可以访问你配置的 Flask 应用。