Upload 6 files

app.py

@@ -0,0 +1,18 @@
+import json
+import fn
+import gradio as gr
+
+def fn_run():
+    return json.dumps(fn.run())
+
+with gr.Blocks() as demo:
+    info = gr.Markdown()
+    run_button = gr.Button(value='run')
+    
+    run_button.click(
+        fn=fn_run,
+        outputs=[info],
+        )
+
+if __name__ == '__main__':
+    demo.launch()

fn.py

@@ -0,0 +1,52 @@
+import os
+import json
+import csv
+import subprocess
+import psutil
+
+def load_model():
+    pass
+
+def run():
+    # WSL内で取得するとホストの値とは異なるので使い物にならない
+    try:
+        # load average per cpu count
+        la = round((psutil.getloadavg()[0] / psutil.cpu_count()) * 100)
+
+        # cpu
+        cpu = round(psutil.cpu_percent(percpu=False))
+
+        # ram
+        ram = psutil.virtual_memory()
+        rama = round(ram.available)
+        ramt = round(ram.total)
+        ramp = 100 - ram.percent
+        ramg = round(ram.available / 1073741824)
+
+        # vram
+        cmd = 'nvidia-smi --query-gpu=utilization.gpu,memory.used,memory.total --format=csv,noheader,nounits'
+        csv = subprocess.run(cmd, shell=True, capture_output=True, text=True).stdout
+        r = csv.split(',')
+        gpu = round(float(r[0]))
+        vrama = round(float(r[1]))
+        vramt = round(float(r[2]))
+        vramp = round(vrama / vramt * 100)
+        vramg = round(float(r[1]) / 1024)
+
+        # network
+        # 1秒ぶんだけ取りたかった
+        #net = psutil.net_io_counters()
+        #ni = round((float(net.bytes_recv) / 1048576) * 100)
+        #no = round((float(net.bytes_sent) / 1048576) * 100)
+
+        res = {
+            'la': la,
+            'cpu': cpu,
+            'ram': ramg,
+            'gpu': gpu,
+            'vram': vramg,
+        }
+
+        return res
+    except Exception as e:
+        return {}

install.bat

@@ -0,0 +1,56 @@
+@echo off
+
+rem -------------------------------------------
+rem NOT guaranteed to work on Windows
+
+set APPDIR=perf
+set REPOS=https://huggingface.co/spaces/aka7774/%APPDIR%
+set VENV=venv
+
+rem -------------------------------------------
+
+set INSTALL_DIR=%~dp0
+cd /d %INSTALL_DIR%
+
+:git_clone
+set DL_URL=%REPOS%
+set DL_DST=%APPDIR%
+git clone %DL_URL% %APPDIR%
+if exist %DL_DST% goto install_python
+
+set DL_URL=https://github.com/git-for-windows/git/releases/download/v2.41.0.windows.3/PortableGit-2.41.0.3-64-bit.7z.exe
+set DL_DST=PortableGit-2.41.0.3-64-bit.7z.exe
+curl -L -o %DL_DST% %DL_URL%
+if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
+%DL_DST% -y
+del %DL_DST%
+
+set GIT=%INSTALL_DIR%PortableGit\bin\git
+%GIT% clone %REPOS%
+
+:install_python
+set DL_URL=https://github.com/indygreg/python-build-standalone/releases/download/20240107/cpython-3.10.13+20240107-i686-pc-windows-msvc-shared-install_only.tar.gz
+set DL_DST="%INSTALL_DIR%python.tar.gz"
+curl -L -o %DL_DST% %DL_URL%
+if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
+tar -xzf %DL_DST%
+
+set PYTHON=%INSTALL_DIR%python\python.exe
+set PATH=%PATH%;%INSTALL_DIR%python310\Scripts
+
+:install_venv
+cd %APPDIR%
+%PYTHON% -m venv %VENV%
+set PYTHON=%VENV%\Scripts\python.exe
+
+:install_pip
+set DL_URL=https://bootstrap.pypa.io/get-pip.py
+set DL_DST=%INSTALL_DIR%get-pip.py
+curl -o %DL_DST% %DL_URL%
+if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
+%PYTHON% %DL_DST%
+
+%PYTHON% -m pip install gradio
+%PYTHON% -m pip install -r requirements.txt
+
+pause

main.py

@@ -0,0 +1,40 @@
+import os
+import sys
+import time
+import signal
+import psutil
+import io
+
+from fastapi import FastAPI, Request, status, Form, UploadFile
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel, Field
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import JSONResponse
+
+import fn
+import gradio as gr
+from app import demo
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=['*'],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+gr.mount_gradio_app(app, demo, path="/gradio")
+
+fn.load_model()
+
+@app.get("/json")
+async def api_json():
+    try:
+        results = fn.run()
+
+        return results
+    except Exception as e:
+        return {"error": str(e)}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+python-multipart
+psutil

venv.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/bash
+
+python3 -m venv venv
+curl -kL https://bootstrap.pypa.io/get-pip.py | venv/bin/python
+
+venv/bin/python -m pip install gradio
+venv/bin/python -m pip install -r requirements.txt