Tuathe commited on
Commit
4ce11fa
·
verified ·
1 Parent(s): 6445700

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- safe_filename = log_text.lower().replace(" ", "_").replace(":", "").replace('"', '').replace(".", "")
70
- runbook_path = f"runbooks/runbook_{safe_filename}.md"
 
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, slack_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))