Hugging Face's logo

Spaces:
imseldrith
/
udemyBot
Running

App Files Community
1
imseldrith commited on
Commit
14b7018
·
verified ·
1 Parent(s): de50ad7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -27
app.py CHANGED
@@ -1,13 +1,10 @@
1
- from flask import Flask, render_template_string
2
- from flask_socketio import SocketIO
3
  from apscheduler.schedulers.background import BackgroundScheduler
4
  import subprocess
5
  import threading
6
  from datetime import datetime
7
 
8
  app = Flask(__name__)
9
- socketio = SocketIO(app, cors_allowed_origins="*") # Enable WebSocket
10
-
11
  execution_logs = []
12
  MAX_LOG_ENTRIES = 20
13
 
@@ -25,25 +22,22 @@ def run_cli_script():
25
  bufsize=1
26
  )
27
 
28
- # Stream logs in real-time to UI
29
  for line in process.stdout:
30
  log_entry['output'] += line
31
- socketio.emit('log_update', {'time': timestamp, 'output': line, 'error': ''}) # Send to UI
32
- print(line, end="") # Print to terminal for debugging
 
33
 
34
  for line in process.stderr:
35
  log_entry['error'] += line
36
- socketio.emit('log_update', {'time': timestamp, 'output': '', 'error': line}) # Send to UI
37
- print(line, end="")
 
38
 
39
  except Exception as e:
40
  log_entry['error'] = str(e)
41
- socketio.emit('log_update', {'time': timestamp, 'output': '', 'error': str(e)}) # Send error to UI
42
-
43
- finally:
44
- execution_logs.append(log_entry)
45
- if len(execution_logs) > MAX_LOG_ENTRIES:
46
- execution_logs.pop(0)
47
 
48
  def start_initial_run():
49
  threading.Thread(target=run_cli_script, daemon=True).start()
@@ -71,18 +65,25 @@ def home():
71
  <html>
72
  <head>
73
  <title>Script Scheduler</title>
74
- <script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js"></script>
75
  <script>
76
- var socket = io.connect(location.protocol + '//' + document.domain + ':' + location.port);
77
- socket.on('log_update', function(data) {
78
- var logBox = document.getElementById("log-box");
79
- var logEntry = "<div class='timestamp'>" + data.time + "</div>";
80
- if (data.output) logEntry += "<div class='output'>" + data.output + "</div>";
81
- if (data.error) logEntry += "<div class='error'>" + data.error + "</div>";
82
- logEntry += "<hr>";
83
- logBox.innerHTML += logEntry; // Append new logs
84
- logBox.scrollTop = logBox.scrollHeight; // Auto-scroll to bottom
85
- });
 
 
 
 
 
 
 
 
86
  </script>
87
  <style>
88
  body { font-family: Arial, sans-serif; padding: 20px; }
@@ -111,6 +112,11 @@ def home():
111
  </html>
112
  ''', next_run=next_run)
113
 
 
 
 
 
 
114
  @app.route('/force-run')
115
  def force_run():
116
  """Manually trigger the script execution."""
@@ -126,4 +132,4 @@ def run_check():
126
  return "Scheduler is running", 200
127
 
128
  if __name__ == '__main__':
129
- socketio.run(app, host='0.0.0.0', port=7860, allow_unsafe_werkzeug=True)
 
1
+ from flask import Flask, render_template_string, jsonify
 
2
  from apscheduler.schedulers.background import BackgroundScheduler
3
  import subprocess
4
  import threading
5
  from datetime import datetime
6
 
7
  app = Flask(__name__)
 
 
8
  execution_logs = []
9
  MAX_LOG_ENTRIES = 20
10
 
 
22
  bufsize=1
23
  )
24
 
25
+ # Stream logs line by line
26
  for line in process.stdout:
27
  log_entry['output'] += line
28
+ execution_logs.append({'time': timestamp, 'output': line, 'error': ''})
29
+ if len(execution_logs) > MAX_LOG_ENTRIES:
30
+ execution_logs.pop(0)
31
 
32
  for line in process.stderr:
33
  log_entry['error'] += line
34
+ execution_logs.append({'time': timestamp, 'output': '', 'error': line})
35
+ if len(execution_logs) > MAX_LOG_ENTRIES:
36
+ execution_logs.pop(0)
37
 
38
  except Exception as e:
39
  log_entry['error'] = str(e)
40
+ execution_logs.append({'time': timestamp, 'output': '', 'error': str(e)})
 
 
 
 
 
41
 
42
  def start_initial_run():
43
  threading.Thread(target=run_cli_script, daemon=True).start()
 
65
  <html>
66
  <head>
67
  <title>Script Scheduler</title>
 
68
  <script>
69
+ function fetchLogs() {
70
+ fetch('/logs')
71
+ .then(response => response.json())
72
+ .then(data => {
73
+ let logBox = document.getElementById("log-box");
74
+ logBox.innerHTML = "";
75
+ data.logs.forEach(log => {
76
+ let logEntry = "<div class='timestamp'>" + log.time + "</div>";
77
+ if (log.output) logEntry += "<div class='output'>" + log.output + "</div>";
78
+ if (log.error) logEntry += "<div class='error'>" + log.error + "</div>";
79
+ logEntry += "<hr>";
80
+ logBox.innerHTML += logEntry;
81
+ });
82
+ logBox.scrollTop = logBox.scrollHeight;
83
+ });
84
+ }
85
+ setInterval(fetchLogs, 2000);
86
+ window.onload = fetchLogs;
87
  </script>
88
  <style>
89
  body { font-family: Arial, sans-serif; padding: 20px; }
 
112
  </html>
113
  ''', next_run=next_run)
114
 
115
+ @app.route('/logs')
116
+ def logs():
117
+ """Returns logs as JSON for AJAX polling."""
118
+ return jsonify({'logs': execution_logs})
119
+
120
  @app.route('/force-run')
121
  def force_run():
122
  """Manually trigger the script execution."""
 
132
  return "Scheduler is running", 200
133
 
134
  if __name__ == '__main__':
135
+ app.run(host='0.0.0.0', port=7860)