Update app.py

app.py

@@ -1,45 +1,216 @@
-from flask import Flask, request, Response
-import requests
+from flask import Flask, request, jsonify, render_template
+from flask_socketio import SocketIO, emit
+import json
+import uuid
+import os
+from datetime import datetime
 
 app = Flask(__name__)
+app.config['SECRET_KEY'] = 'your-secret-key'
+socketio = SocketIO(app)
 
-# HTML form template to allow user input
-html_form = '''
-    <form action="/proxy" method="get">
-        <label for="url">Enter URL:</label>
-        <input type="text" id="url" name="url" placeholder="https://example.com" required>
-        <button type="submit">Browse</button>
-    </form>
-'''
+# Job storage with unique IDs and status
+jobs = []
+connected_clients = {}
+
+def save_jobs():
+    """Save jobs to JSON file"""
+    try:
+        with open('jobs.json', 'w') as f:
+            json.dump(jobs, f)
+    except Exception as e:
+        print(f"Error saving jobs: {e}")
+
+def load_jobs():
+    """Load jobs from JSON file"""
+    try:
+        if os.path.exists('jobs.json') and os.path.getsize('jobs.json') > 0:
+            with open('jobs.json', 'r') as f:
+                return json.load(f)
+    except Exception as e:
+        print(f"Error loading jobs: {e}")
+    return []
 
 @app.route('/')
-def home():
-    # Display the form for URL input
-    return html_form
+def index():
+    return render_template('index.html')
 
-@app.route('/proxy')
-def proxy():
-    # Get the URL from the user's input
-    url = request.args.get('url')
-    if not url:
-        return "No URL provided", 400
+@app.route('/submit_job', methods=['POST'])
+def submit_job():
+    youtube_url = request.form.get("youtube_url")
+    exit_time = int(request.form.get("exit_time", 10))
+    mode = request.form.get("mode", "once")
+    job_id = str(uuid.uuid4())
 
-    # Validate URL (must start with http:// or https:// for security)
-    if not (url.startswith("http://") or url.startswith("https://")):
-        return "Invalid URL. Please include http:// or https://", 400
+    job = {
+        "job_id": job_id,
+        "youtube_url": youtube_url,
+        "exit_time": exit_time,
+        "mode": mode,
+        "status": "pending"
+    }
+    jobs.append(job)
+    save_jobs()
 
-    try:
-        # Fetch the content from the specified URL
-        response = requests.get(url)
-        response.raise_for_status()  # Ensure the request was successful
+    return jsonify({"message": "Job submitted successfully", "job_id": job_id})
+
+@socketio.on('client_info')
+def handle_client_info(data):
+    client_id = request.sid
+    connected_clients[client_id] = {
+        'ip': data.get('ip', 'Unknown'),
+        'country': data.get('country', 'Unknown'),
+        'city': data.get('city', 'Unknown'),
+        'connected_at': datetime.now().isoformat()
+    }
+    
+    emit('client_connected', {
+        'client_id': client_id,
+        'ip': data.get('ip', 'Unknown'),
+        'country': data.get('country', 'Unknown'),
+        'city': data.get('city', 'Unknown')
+    }, broadcast=True)
+
+@socketio.on('disconnect')
+def handle_disconnect():
+    client_id = request.sid
+    if client_id in connected_clients:
+        del connected_clients[client_id]
+        emit('client_disconnected', client_id, broadcast=True)
+
+@app.route('/get_job', methods=['GET'])
+def get_job():
+    for job in jobs:
+        if job["status"] == "pending":
+            return jsonify(job)
+    return jsonify({"message": "No pending jobs"})
 
-        # Return the fetched content as a response to the user
-        return Response(response.content, content_type=response.headers['Content-Type'])
+@app.route('/update_job_status', methods=['POST'])
+def update_job_status():
+    data = request.get_json()
+    job_id = data.get("job_id")
+    status = data.get("status", "done")
 
-    except requests.RequestException as e:
-        # Handle errors in fetching the page
-        return f"Error fetching the page: {e}"
+    for job in jobs:
+        if job["job_id"] == job_id:
+            job["status"] = status
+            break
+
+    save_jobs()
+    return jsonify({"message": f"Job {job_id} marked as {status}"})
 
 if __name__ == "__main__":
-    # Run the Flask app on port 7860
-    app.run(host="0.0.0.0", port=7860)
+    jobs = load_jobs()
+    
+    if not os.path.exists('templates'):
+        os.makedirs('templates')
+    
+    template_path = os.path.join('templates', 'index.html')
+    if not os.path.exists(template_path):
+        with open(template_path, 'w') as f:
+            f.write('''<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>YouTube Job Manager</title>
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css" rel="stylesheet">
+    <style>
+        body { padding: 20px; }
+        .connection-status { margin-bottom: 20px; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="row">
+            <div class="col-md-8 offset-md-2">
+                <h1 class="mb-4">YouTube Job Manager</h1>
+                
+                <div class="card mb-4">
+                    <div class="card-header">
+                        <h5 class="mb-0">Connected Clients</h5>
+                    </div>
+                    <div class="card-body">
+                        <table class="table" id="clientsTable">
+                            <thead>
+                                <tr>
+                                    <th>IP Address</th>
+                                    <th>Country</th>
+                                    <th>City</th>
+                                    <th>Status</th>
+                                </tr>
+                            </thead>
+                            <tbody id="clientsTableBody">
+                            </tbody>
+                        </table>
+                    </div>
+                </div>
+
+                <div class="card">
+                    <div class="card-header">
+                        <h5 class="mb-0">Submit New Job</h5>
+                    </div>
+                    <div class="card-body">
+                        <form action="/submit_job" method="post" class="needs-validation" novalidate>
+                            <div class="mb-3">
+                                <label for="youtube_url" class="form-label">YouTube URL:</label>
+                                <input type="text" class="form-control" id="youtube_url" name="youtube_url" required>
+                            </div>
+
+                            <div class="mb-3">
+                                <label for="exit_time" class="form-label">Exit Time (seconds):</label>
+                                <input type="number" class="form-control" id="exit_time" name="exit_time" value="10" required>
+                            </div>
+
+                            <div class="mb-3">
+                                <label for="mode" class="form-label">Mode:</label>
+                                <select class="form-select" id="mode" name="mode">
+                                    <option value="once">Once</option>
+                                    <option value="again">Again</option>
+                                    <option value="continuous">Continuous</option>
+                                </select>
+                            </div>
+
+                            <button type="submit" class="btn btn-primary">Submit Job</button>
+                        </form>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </div>
+
+    <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/js/bootstrap.bundle.min.js"></script>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js"></script>
+    <script>
+        const socket = io();
+        
+        socket.on('connect', () => {
+            console.log('Connected to server');
+        });
+
+        socket.on('client_connected', (data) => {
+            const tableBody = document.getElementById('clientsTableBody');
+            const row = document.createElement('tr');
+            row.setAttribute('data-client-id', data.client_id);
+            row.innerHTML = `
+                <td>${data.ip}</td>
+                <td>${data.country}</td>
+                <td>${data.city}</td>
+                <td><span class="badge bg-success">Connected</span></td>
+            `;
+            tableBody.appendChild(row);
+        });
+
+        socket.on('client_disconnected', (clientId) => {
+            const row = document.querySelector(`tr[data-client-id="${clientId}"]`);
+            if (row) {
+                row.querySelector('.badge').className = 'badge bg-danger';
+                row.querySelector('.badge').textContent = 'Disconnected';
+            }
+        });
+    </script>
+</body>
+</html>
+''')
+    
+    socketio.run(app, host="0.0.0.0", port=7860 debug=False)