admin commited on
Commit
e8e1626
·
1 Parent(s): 7d041c5

add manual trigger

Browse files
Files changed (2) hide show
  1. activate.py +4 -1
  2. app.py +18 -7
activate.py CHANGED
@@ -147,10 +147,11 @@ def trigger(users=USERS):
147
  threading.Thread(target=activate_studio, args=(studio,), daemon=True).start()
148
  time.sleep(DELAY)
149
 
150
- print(
151
  "\n".join(spaces + studios)
152
  + f"\n[{fix_datetime(datetime.now())}] Activation complete!"
153
  )
 
154
  content = ""
155
  for failure in failures:
156
  errepo: str = failure
@@ -159,3 +160,5 @@ def trigger(users=USERS):
159
 
160
  if content:
161
  send_email(content)
 
 
 
147
  threading.Thread(target=activate_studio, args=(studio,), daemon=True).start()
148
  time.sleep(DELAY)
149
 
150
+ log = (
151
  "\n".join(spaces + studios)
152
  + f"\n[{fix_datetime(datetime.now())}] Activation complete!"
153
  )
154
+ print(log)
155
  content = ""
156
  for failure in failures:
157
  errepo: str = failure
 
160
 
161
  if content:
162
  send_email(content)
163
+
164
+ return log
app.py CHANGED
@@ -40,10 +40,21 @@ def tasklist():
40
 
41
  if __name__ == "__main__":
42
  monitor()
43
- gr.Interface(
44
- title="See current task status",
45
- fn=tasklist,
46
- inputs=None,
47
- outputs=gr.TextArea(label="Current task details"),
48
- flagging_mode="never",
49
- ).launch(ssr_mode=False)
 
 
 
 
 
 
 
 
 
 
 
 
40
 
41
  if __name__ == "__main__":
42
  monitor()
43
+ with gr.Blocks() as demo:
44
+ gr.Interface(
45
+ title="See current task status",
46
+ fn=tasklist,
47
+ inputs=None,
48
+ outputs=gr.TextArea(label="Current task details"),
49
+ flagging_mode="never",
50
+ )
51
+
52
+ gr.Interface(
53
+ title="Trigger once manually",
54
+ fn=trigger,
55
+ inputs=None,
56
+ outputs=gr.TextArea(label="Activation logs"),
57
+ flagging_mode="never",
58
+ )
59
+
60
+ demo.launch()