@FIR-748: Added endpoints for health, sysinfo, upload and restart (#15)

Author: atrivedi-tsavoritesi

tools/flaskIfc/flaskIfc.py

@@ -2,25 +2,24 @@
 import subprocess
 import threading
 import time
+from werkzeug.utils import secure_filename
+import os
 
 job_status = {"running": False, "result": "", "thread": None}
 
 app = Flask(__name__)
 
+port = '/dev/ttyUSB3'
+baudrate = '921600'
+
 @app.route('/')
 def index():
     return render_template('index.html')
 
 @app.route('/llama-cli', methods=['GET'])
-def serial_command():
-    # Currently the port is hard coded to /dev/ttyUSB3 but can be parameterized
-    port = '/dev/ttyUSB3'
-    #port = request.args.get('port')
-
-    # Currently the baudrate is hard coded to 921600 but can be parameterized
-    #baudrate = request.args.get('baudrate')
-    baudrate = '921600'
-    #./run_platform_test.sh "my cat's name" "10" "tinyllama-vo-5m-para.gguf" "none"
+def llama_cli_serial_command():
+
+    #./run_llama_cli.sh "my cat's name" "10" "tinyllama-vo-5m-para.gguf" "none"
     model = request.args.get('model')
     backend = request.args.get('backend')
     tokens = request.args.get('tokens')
@@ -59,7 +58,95 @@ def serial_command():
     except subprocess.CalledProcessError as e:
         return f"Error executing script: {e.stderr}", 500
 
+UPLOAD_FOLDER = './' # Directory where uploaded files will be stored
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+os.makedirs(UPLOAD_FOLDER, exist_ok=True) # Create the upload folder if it doesn't exist
+
+@app.route('/upload-gguf', methods=['POST', 'GET'])
+def upload_serial_command():
+    if request.method == 'POST':
+        # Check if a file was submitted
+        if 'file' not in request.files:
+            return "No file part"
+        file = request.files['file']
+
+        # Check if the file is empty
+        if file.filename == '':
+            return "No file selected"
+
+       # Save the file if it exists
+        if file:
+            filename = secure_filename(file.filename)
+            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+            return "File uploaded successfully"
+    return render_template('upload.html') # Display the upload form
+
+#    command = f"upload file"
+#    try:
+#        result = subprocess.run(['python3', 'serial_script.py', port, baudrate, command], capture_output=True, text=True, check=True)
+#        return result.stdout, 200
+#    except subprocess.CalledProcessError as e:
+#        return f"Error executing script: {e.stderr}", 500
+
+@app.route('/upload-file', methods=['GET', 'POST'])
+def upload_file():
+    if request.method == 'POST':
+        # Check if a file was submitted
+        if 'file' not in request.files:
+            return "No file part"
+        file = request.files['file']
+
+        # Check if the file is empty
+        if file.filename == '':
+            return "No file selected"
+
+        # Save the file if it exists
+        if file:
+            filename = secure_filename(file.filename)
+            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+            return "File uploaded successfully"
+    return render_template('upload.html') # Display the upload form
+
+@app.route('/restart-txe', methods=['GET'])
+def restart_txe_serial_command():
+    command = f"telnet localhost 8000; close all"
+
+    try:
+        result = subprocess.run(['python3', 'serial_script.py', port, baudrate, command], capture_output=True, text=True, check=True)
+        return result.stdout, 200
+    except subprocess.CalledProcessError as e:
+        return f"Error executing script: {e.stderr}", 500
+
+@app.route('/health-check', methods=['GET'])
+def health_check_serial_command():
+    command = f"free -h"
+
+    try:
+        result = subprocess.run(['python3', 'serial_script.py', port, baudrate, command], capture_output=True, text=True, check=True)
+        return result.stdout, 200
+    except subprocess.CalledProcessError as e:
+        return f"Error executing script: {e.stderr}", 500
 
+@app.route('/test', methods=['GET'])
+def test_serial_command():
+    command = f"test"
+
+    try:
+        result = subprocess.run(['python3', 'serial_script.py', port, baudrate, command], capture_output=True, text=True, check=True)
+        return result.stdout, 200
+    except subprocess.CalledProcessError as e:
+        return f"Error executing script: {e.stderr}", 500
+
+@app.route('/system-info', methods=['GET'])
+def system_info_serial_command():
+
+    command = f"lscpu"
+
+    try:
+        result = subprocess.run(['python3', 'serial_script.py', port, baudrate, command], capture_output=True, text=True, check=True)
+        return result.stdout, 200
+    except subprocess.CalledProcessError as e:
+        return f"Error executing script: {e.stderr}", 500
 
 @app.route('/submit', methods=['POST'])
 def submit():
@@ -68,7 +155,7 @@ def submit():
     if job_status["running"]:
         return "<h2>A model is already running. Please wait or abort.</h2>"
 
-    #./run_platform_test.sh "my cat's name" "10" "tinyllama-vo-5m-para.gguf" "none"
+    #./run_llama_cli.sh "my cat's name" "10" "tinyllama-vo-5m-para.gguf" "none"
     model = request.form.get('model')
     backend = request.form.get('backend')
     tokens = request.form.get('tokens')
@@ -96,11 +183,7 @@ def submit():
     #    "--top-k", "0",
     #    "--top-p", "1"
     #]
-    # Currently the port is hard coded to /dev/ttyUSB3 but can be parameterized
-    port = '/dev/ttyUSB3'
 
-    # Currently the baudrate is hard coded to 921600 but can be parameterized
-    baudrate = '921600'
     script_path = "/usr/bin/tsi/v0.1.1.tsv31_06_06_2025/bin/run_llama_cli.sh"
     command = f"{script_path} \"{prompt}\" {tokens} {model_path} {backend}"
 

tools/flaskIfc/templates/upload.html

@@ -0,0 +1,4 @@
+<form method="post" enctype="multipart/form-data">
+    <input type="file" name="file">
+    <input type="submit" value="Upload">
+</form>