Spaces:
Runtime error
Runtime error
Update streamlit_app.py
Browse files- streamlit_app.py +5 -3
streamlit_app.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import streamlit as st
|
| 2 |
import os
|
|
|
|
| 3 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 4 |
import torch
|
| 5 |
from upload_to_s3 import upload_file_to_s3
|
|
@@ -66,8 +67,9 @@ Only output valid markdown text.
|
|
| 66 |
outputs = model.generate(**inputs, max_new_tokens=300)
|
| 67 |
runbook_text = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
|
| 68 |
|
| 69 |
-
|
| 70 |
-
runbook_path = f"runbooks/runbook_{
|
|
|
|
| 71 |
os.makedirs("runbooks", exist_ok=True)
|
| 72 |
with open(runbook_path, "w", encoding="utf-8") as f:
|
| 73 |
f.write(runbook_text)
|
|
@@ -106,7 +108,7 @@ if st.button(" Classify + Generate Runbook"):
|
|
| 106 |
|
| 107 |
if success:
|
| 108 |
s3_url = f"https://{bucket_name}.s3.{aws_region}.amazonaws.com/{s3_path}"
|
| 109 |
-
send_slack_alert(log, s3_url
|
| 110 |
st.success(" Slack alert sent.")
|
| 111 |
st.markdown(f"[ View Runbook on S3]({s3_url})")
|
| 112 |
st.download_button(" Download Runbook", runbook_md, file_name=os.path.basename(runbook_path))
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
import os
|
| 3 |
+
import hashlib
|
| 4 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 5 |
import torch
|
| 6 |
from upload_to_s3 import upload_file_to_s3
|
|
|
|
| 67 |
outputs = model.generate(**inputs, max_new_tokens=300)
|
| 68 |
runbook_text = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
|
| 69 |
|
| 70 |
+
hash_id = hashlib.md5(log_text.encode()).hexdigest()[:8]
|
| 71 |
+
runbook_path = f"runbooks/runbook_{hash_id}.md"
|
| 72 |
+
|
| 73 |
os.makedirs("runbooks", exist_ok=True)
|
| 74 |
with open(runbook_path, "w", encoding="utf-8") as f:
|
| 75 |
f.write(runbook_text)
|
|
|
|
| 108 |
|
| 109 |
if success:
|
| 110 |
s3_url = f"https://{bucket_name}.s3.{aws_region}.amazonaws.com/{s3_path}"
|
| 111 |
+
send_slack_alert(log, s3_url)
|
| 112 |
st.success(" Slack alert sent.")
|
| 113 |
st.markdown(f"[ View Runbook on S3]({s3_url})")
|
| 114 |
st.download_button(" Download Runbook", runbook_md, file_name=os.path.basename(runbook_path))
|