add files for cs2

automated_deployment

@@ -1,48 +0,0 @@
-#! /bin/bash
-
-PORT=21003
-MACHINE=paffenroth-23.dyn.wpi.edu
-STUDENT_ADMIN_KEY_PATH=kfkey
-
-#insert key into list of authorized keys
-cat kfkey.pub > authorized_keys
-
-#set permissions
-chmod 600 authorized_keys
-
-#remove default key
-rm -f student-admin_key*
-
-# Copy the authorized_keys file to the server
-scp -i student-admin_key -P ${PORT} -o StrictHostKeyChecking=no authorized_keys student-admin@${MACHINE}:~/.ssh/
-
-# Add the key to the ssh-agent
-eval "$(ssh-agent -s)"
-ssh-add mykey
-
-# Check the key file on the server
-echo "checking that the authorized_keys file is correct"
-ssh -p ${PORT} -o StrictHostKeyChecking=no student-admin@${MACHINE} "cat ~/.ssh/authorized_keys"
-
-# clone the repo
-git clone https://github.com/rcpaffenroth/CS553_example
-
-# Copy the files to the server
-scp -P ${PORT} -o StrictHostKeyChecking=no -r CS553_example student-admin@${MACHINE}:~/
-
-# check that the code in installed and start up the product
-# COMMAND="ssh -p ${PORT} -o StrictHostKeyChecking=no student-admin@${MACHINE}"
-
-# ${COMMAND} "ls CS553_example"
-# ${COMMAND} "sudo apt install -qq -y python3-venv"
-# ${COMMAND} "cd CS553_example && python3 -m venv venv"
-# ${COMMAND} "cd CS553_example && source venv/bin/activate && pip install -r requirements.txt"
-# ${COMMAND} "nohup CS553_example/venv/bin/python3 CS553_example/app.py > log.txt 2>&1 &"
-
-# nohup ./whatever > /dev/null 2>&1 
-
-# debugging ideas
-# sudo apt-get install gh
-# gh auth login
-# requests.exceptions.HTTPError: 429 Client Error: Too Many Requests for url: https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta/v1/chat/completions
-# log.txt

automated_deployment.sh

@@ -0,0 +1,22 @@
+#! /bin/bash
+
+# Load environment variables from .env file
+if [ -f .env ]; then
+  source .env  # loads the variables into current shell session
+else
+  echo ".env file not found"
+  exit 1
+fi
+
+# ssh into the vm
+# check that the code in installed and start up the product
+COMMAND="ssh -i ${STUDENT_ADMIN_KEY_PATH} -p ${PORT} -o StrictHostKeyChecking=no student-admin@${MACHINE}"
+
+# clone the repo
+${COMMAND} "git clone https://github.com/jcai0o0/cs553-case-study-1.git"
+${COMMAND} "sudo apt install -qq -y python3-venv"
+${COMMAND} "cd cs553-case-study-1 && python3 -m venv cs2_venv"
+${COMMAND} "cd cs553-case-study-1 && source cs2_venv/bin/activate && pip install -r requirements.txt"
+${COMMAND} "cd cs553-case-study-1 && source cs2_venv/bin/activate && pip install -r requirements.txt"
+${COMMAND} "nohup cs553-case-study-1/cs2_venv/bin/python3 cs553-case-study-1/app.py > log.txt 2>&1 &"
+

requirements.txt

@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 huggingface_hub==0.23.*
-gradio==4.39.*
+gradio==4.43.0
 torch==2.4.*
 transformers==4.43.*
 accelerate==0.33.*

watcher.py

@@ -0,0 +1,104 @@
+import os
+import requests
+import time
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from dotenv import load_dotenv
+import subprocess
+
+load_dotenv()
+
+# Settings
+VM_IP = "paffenroth-23.dyn.wpi.edu"
+GRADIO_UI_URL = "http://paffenroth-23.dyn.wpi.edu:8005/"
+CHECK_INTERVAL = 60   # 60 seconds
+RECOVERY_SCRIPT_PATH = 'automated_deployment.sh'
+
+
+# Email settings (if EMAIL_ALERT is True)
+EMAIL_SENDER = os.getenv("EMAIL_SENDER")
+EMAIL_RECEIVER = os.getenv("EMAIL_RECEIVER")
+EMAIL_PASSWORD = os.getenv("EMAIL_PASSWORD")
+
+
+def send_email_alert(subject, body):
+    msg = MIMEMultipart()
+    msg['From'] = EMAIL_SENDER
+    msg['To'] = EMAIL_RECEIVER
+    msg['Subject'] = subject
+
+    msg.attach(MIMEText(body, 'plain'))
+
+    try:
+        with smtplib.SMTP('smtp.gmail.com', 587) as server:
+            server.starttls()
+            server.login(EMAIL_SENDER, EMAIL_PASSWORD)
+            text = msg.as_string()
+            server.sendmail(EMAIL_SENDER, EMAIL_RECEIVER, text)
+        print("Email alert sent successfully!")
+    except Exception as e:
+        print(f"Failed to send email alert: {e}")
+
+
+def check_vm_status():
+    response = os.system(f"ping -c 1 {VM_IP}")  # send one ICMP echo request
+    return response == 0  # successful ping
+
+
+def check_ui_status():
+    try:
+        response = requests.get(GRADIO_UI_URL)
+        if response.status_code == 200:
+            return True
+        else:
+            print(f"UI returned status code {response.status_code}")
+            return False
+    except requests.ConnectionError:
+        print("Failed to connect to UI.")
+        return False
+
+
+def monitor_vm(EMAIL_ALERT = False):
+    while True:
+        ui_status = check_ui_status()
+        # vm_status = check_vm_status()
+        if ui_status:  # and vm_status:
+            print(f"{time.ctime()} ---- Meowthematical chatbot is up and running")
+        else:
+            if not ui_status:
+                print(f"{time.ctime()} ---- PRODUCT is DOWN!!!")
+                if check_vm_status():  # check if VM is up and running
+                    print(f"{time.ctime()} ---- VM {VM_IP} is up and running, starting recovery process...")
+                    try:
+                        result = subprocess.run(['bash', RECOVERY_SCRIPT_PATH],
+                                                check=True,
+                                                capture_output=True,
+                                                text=True)
+                        # Output from the script
+                        # print("Output of the script:")
+                        # print(result.stdout)  # Standard output of the script
+                        # print("Errors (if any):")
+                        # print(result.stderr)  # Standard error output of the script
+                    except subprocess.CalledProcessError as e:
+                        print(f"An error occurred: {e}")
+                        print(f"Return code: {e.returncode}")
+                        print(f"Output: {e.output}")
+                        print(f"Error: {e.stderr}")
+
+                    if EMAIL_ALERT:
+                        send_email_alert(
+                            subject="MEOWTHMATICAL Down Alert",
+                            body=f"MEOWTHEMATICAL with IP {VM_IP} is down as of {time.ctime()}"
+                        )
+                else:  # both product and VM is gone
+                    if EMAIL_ALERT:
+                        send_email_alert(
+                            subject="VM Down Alert",
+                            body=f"Vitural Machine at IP {VM_IP} is down as of {time.ctime()}"
+                        )
+        time.sleep(CHECK_INTERVAL)
+
+
+if __name__ == "__main__":
+    monitor_vm(EMAIL_ALERT=False)