Spaces:
Running
Running
| from flask import Flask, request, redirect, url_for, render_template_string | |
| import subprocess | |
| import threading | |
| app = Flask(__name__) | |
| # Replace these with your desired username and password | |
| USERNAME = 'your_username' | |
| PASSWORD = 'your_password' | |
| html_login = ''' | |
| <!doctype html> | |
| <title>Login</title> | |
| <h1>Login</h1> | |
| <form method=post> | |
| <input type=text name=username placeholder='Username'> | |
| <input type=password name=password placeholder='Password'> | |
| <input type=submit value=Login> | |
| </form> | |
| ''' | |
| def login(): | |
| if request.method == 'POST': | |
| if request.form['username'] == USERNAME and request.form['password'] == PASSWORD: | |
| return redirect(url_for('gradio_app')) | |
| else: | |
| return 'Incorrect credentials', 401 | |
| return render_template_string(html_login) | |
| def gradio_app(): | |
| # This will ensure the app runs only once and not on every request | |
| if not gradio_app.started: | |
| threading.Thread(target=run_gradio).start() | |
| gradio_app.started = True | |
| return redirect("http://localhost:7860") | |
| def run_gradio(): | |
| subprocess.run(["python", "app_w_lora.py"]) | |
| # Initialize the flag to indicate if the Gradio app has started | |
| gradio_app.started = False | |
| if __name__ == "__main__": | |
| app.run(host='0.0.0.0', port=5000) | |