Create app.py

app.py

@@ -0,0 +1,104 @@
+import os
+import gradio as gr
+import threading
+from flask import Flask
+from psycopg2 import connect
+
+# Fetch environment variables
+DB_NAME = os.getenv("DB_NAME")
+DB_USER = os.getenv("DB_USER")
+DB_PASSWORD = os.getenv("DB_PASSWORD")
+DB_HOST = os.getenv("DB_HOST")
+DB_PORT = os.getenv("DB_PORT")
+APP_PASSWORD = os.getenv("APP_PASSWORD")
+
+# Database connection
+def get_db_connection():
+    return connect(
+        dbname=DB_NAME,
+        user=DB_USER,
+        password=DB_PASSWORD,
+        host=DB_HOST,
+        port=DB_PORT
+    )
+
+# Dictionary to manage server threads
+server_threads = {}
+
+# Flask app template for HTTP servers
+def create_server(name, port):
+    app = Flask(name)
+
+    @app.route("/")
+    def home():
+        return f"Server {name} is running on port {port}!"
+
+    return app
+
+# Function to start a server
+def start_server(name, port):
+    if name in server_threads:
+        return f"Server {name} is already running!"
+    
+    app = create_server(name, port)
+
+    def run():
+        app.run(host="0.0.0.0", port=port, debug=False, use_reloader=False)
+
+    thread = threading.Thread(target=run, daemon=True)
+    thread.start()
+    server_threads[name] = thread
+
+    # Update database status
+    conn = get_db_connection()
+    with conn.cursor() as cursor:
+        cursor.execute(
+            "INSERT INTO servers (name, port, status) VALUES (%s, %s, %s) ON CONFLICT (name) DO UPDATE SET status = EXCLUDED.status",
+            (name, port, "running")
+        )
+        conn.commit()
+    conn.close()
+
+    return f"Server {name} started on port {port}!"
+
+# Function to stop a server (dummy implementation for now)
+def stop_server(name):
+    if name not in server_threads:
+        return f"Server {name} is not running!"
+
+    # Terminate the thread (not directly supported, requires external handling)
+    server_threads.pop(name)
+    return f"Server {name} stopped!"
+
+# Gradio UI
+def manage_servers(password, action, name, port):
+    if password != APP_PASSWORD:
+        return "Invalid password!"
+
+    if action == "start":
+        return start_server(name, port)
+    elif action == "stop":
+        return stop_server(name)
+    else:
+        return "Invalid action!"
+
+# Gradio Interface
+with gr.Blocks() as server_manager:
+    gr.Markdown("# Server Manager")
+    with gr.Row():
+        password = gr.Textbox(label="Password", type="password")
+    with gr.Row():
+        action = gr.Radio(["start", "stop"], label="Action")
+        name = gr.Textbox(label="Server Name")
+        port = gr.Number(label="Port")
+    with gr.Row():
+        submit = gr.Button("Submit")
+    output = gr.Textbox(label="Output")
+
+    submit.click(
+        manage_servers,
+        inputs=[password, action, name, port],
+        outputs=output
+    )
+
+server_manager.launch()