Compare commits
11 Commits
e823d59821
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c7ed88ef1f | ||
|
|
b8db4b0882 | ||
|
|
21e2a7747d | ||
|
|
c13dfa5b83 | ||
|
|
22a16affa5 | ||
|
|
c1db6ae562 | ||
|
|
dfcf729355 | ||
| dc25b8780c | |||
|
|
48c3b33bf4 | ||
|
|
4cca68e12b | ||
|
|
658ef42e72 |
20
.woodpecker/check.yml
Normal file
20
.woodpecker/check.yml
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
when:
|
||||||
|
- event: pull_request
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: fmt
|
||||||
|
image: rust:1.84
|
||||||
|
commands:
|
||||||
|
- rustup component add rustfmt
|
||||||
|
- cargo fmt -- --check
|
||||||
|
|
||||||
|
- name: clippy
|
||||||
|
image: rust:1.84
|
||||||
|
commands:
|
||||||
|
- rustup component add clippy
|
||||||
|
- cargo clippy -- -D warnings
|
||||||
|
|
||||||
|
- name: test
|
||||||
|
image: rust:1.84
|
||||||
|
commands:
|
||||||
|
- cargo test
|
||||||
954
Cargo.lock
generated
954
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
@@ -4,5 +4,8 @@ version = "0.1.0"
|
|||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
axum = "0.7"
|
axum = { version = "0.7", features = ["multipart"] }
|
||||||
|
reqwest = { version = "0.12", default-features = false, features = ["multipart", "json", "rustls-tls"] }
|
||||||
|
serde = { version = "1", features = ["derive"] }
|
||||||
|
serde_json = "1"
|
||||||
tokio = { version = "1", features = ["full"] }
|
tokio = { version = "1", features = ["full"] }
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
# Этап сборки
|
# Этап сборки
|
||||||
FROM rust:1.84 as builder
|
FROM rust:1.85 as builder
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
COPY . .
|
COPY . .
|
||||||
RUN cargo build --release
|
RUN cargo build --release
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ spec:
|
|||||||
- name: gitea-registry
|
- name: gitea-registry
|
||||||
containers:
|
containers:
|
||||||
- name: bcard
|
- name: bcard
|
||||||
image: git.mikhailkilin.ru/killingdruid/bcard:fc31b9fd8861debe3aabe473408b1238e858303c
|
image: git.mikhailkilin.ru/killingdruid/bcard:b8db4b08828e27c8b7cad4eedb59f855e43eb584
|
||||||
ports:
|
ports:
|
||||||
- containerPort: 3000
|
- containerPort: 3000
|
||||||
---
|
---
|
||||||
|
|||||||
30
src/main.rs
30
src/main.rs
@@ -1,9 +1,14 @@
|
|||||||
use axum::{response::Html, routing::get, Router};
|
mod transcribe;
|
||||||
|
|
||||||
|
use axum::{response::Html, routing::{get, post}, Router};
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let app = Router::new().route("/", get(handler));
|
let app = Router::new()
|
||||||
|
.route("/", get(handler))
|
||||||
|
.route("/transcribator", get(transcribator_page))
|
||||||
|
.route("/api/transcribe", post(transcribe::transcribe));
|
||||||
let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
|
let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
|
||||||
println!("listening on {}", addr);
|
println!("listening on {}", addr);
|
||||||
let listener = tokio::net::TcpListener::bind(addr).await.unwrap();
|
let listener = tokio::net::TcpListener::bind(addr).await.unwrap();
|
||||||
@@ -13,3 +18,24 @@ async fn main() {
|
|||||||
async fn handler() -> Html<&'static str> {
|
async fn handler() -> Html<&'static str> {
|
||||||
Html("<h1>Mikhail Kilin</h1>")
|
Html("<h1>Mikhail Kilin</h1>")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn transcribator_page() -> Html<&'static str> {
|
||||||
|
Html(include_str!("../static/transcribator.html"))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handler() {
|
||||||
|
let response = handler().await;
|
||||||
|
assert!(response.0.contains("Mikhail Kilin"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_transcribator_page() {
|
||||||
|
let response = transcribator_page().await;
|
||||||
|
assert!(response.0.contains("Transcribator"));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
78
src/transcribe.rs
Normal file
78
src/transcribe.rs
Normal file
@@ -0,0 +1,78 @@
|
|||||||
|
use axum::{extract::Multipart, http::StatusCode, Json};
|
||||||
|
use serde::Serialize;
|
||||||
|
|
||||||
|
const WHISPER_URL: &str = "http://whisper.whisper.svc:8000/v1/audio/transcriptions";
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
pub struct TranscribeResponse {
|
||||||
|
pub text: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn transcribe(
|
||||||
|
mut multipart: Multipart,
|
||||||
|
) -> Result<Json<TranscribeResponse>, (StatusCode, String)> {
|
||||||
|
let mut audio_data: Option<(Vec<u8>, String)> = None;
|
||||||
|
|
||||||
|
while let Some(field) = multipart
|
||||||
|
.next_field()
|
||||||
|
.await
|
||||||
|
.map_err(|e| (StatusCode::BAD_REQUEST, e.to_string()))?
|
||||||
|
{
|
||||||
|
if field.name() == Some("audio") {
|
||||||
|
let file_name = field
|
||||||
|
.file_name()
|
||||||
|
.unwrap_or("recording.webm")
|
||||||
|
.to_string();
|
||||||
|
let data = field
|
||||||
|
.bytes()
|
||||||
|
.await
|
||||||
|
.map_err(|e| (StatusCode::BAD_REQUEST, e.to_string()))?;
|
||||||
|
audio_data = Some((data.to_vec(), file_name));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let (data, file_name) = audio_data
|
||||||
|
.ok_or((StatusCode::BAD_REQUEST, "No audio field".to_string()))?;
|
||||||
|
|
||||||
|
let mime = if file_name.ends_with(".m4a") || file_name.ends_with(".mp4") {
|
||||||
|
"audio/mp4"
|
||||||
|
} else {
|
||||||
|
"audio/webm"
|
||||||
|
};
|
||||||
|
|
||||||
|
let part = reqwest::multipart::Part::bytes(data)
|
||||||
|
.file_name(file_name)
|
||||||
|
.mime_str(mime)
|
||||||
|
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))?;
|
||||||
|
|
||||||
|
let form = reqwest::multipart::Form::new()
|
||||||
|
.part("file", part)
|
||||||
|
.text("model", "Systran/faster-whisper-medium")
|
||||||
|
.text("language", "ru");
|
||||||
|
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
let resp = client
|
||||||
|
.post(WHISPER_URL)
|
||||||
|
.multipart(form)
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.map_err(|e| (StatusCode::BAD_GATEWAY, format!("Whisper unavailable: {e}")))?;
|
||||||
|
|
||||||
|
if !resp.status().is_success() {
|
||||||
|
let status = resp.status();
|
||||||
|
let body = resp.text().await.unwrap_or_default();
|
||||||
|
return Err((StatusCode::BAD_GATEWAY, format!("Whisper {status}: {body}")));
|
||||||
|
}
|
||||||
|
|
||||||
|
let whisper_resp: serde_json::Value = resp
|
||||||
|
.json()
|
||||||
|
.await
|
||||||
|
.map_err(|e| (StatusCode::BAD_GATEWAY, format!("Invalid Whisper response: {e}")))?;
|
||||||
|
|
||||||
|
let text = whisper_resp["text"]
|
||||||
|
.as_str()
|
||||||
|
.unwrap_or("")
|
||||||
|
.to_string();
|
||||||
|
|
||||||
|
Ok(Json(TranscribeResponse { text }))
|
||||||
|
}
|
||||||
118
static/transcribator.html
Normal file
118
static/transcribator.html
Normal file
@@ -0,0 +1,118 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="ru" data-theme="dark">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Transcribator</title>
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@picocss/pico@2/css/pico.min.css">
|
||||||
|
<style>
|
||||||
|
:root { --pico-font-size: 16px; }
|
||||||
|
html, body { height: 100dvh; margin: 0; overflow: hidden; }
|
||||||
|
body { display: flex; flex-direction: column; }
|
||||||
|
main { flex: 1; display: flex; flex-direction: column; max-width: 640px; width: 100%; margin: 0 auto; padding: 0.5rem; overflow: hidden; }
|
||||||
|
h1 { text-align: center; margin: 0.5rem 0; font-size: 1.4rem; }
|
||||||
|
#chat { flex: 1; overflow-y: auto; display: flex; flex-direction: column; gap: 0.75rem; padding: 0.5rem 0; }
|
||||||
|
#chat:empty::before { content: "Tap the microphone to start recording"; display: block; text-align: center; opacity: 0.5; margin-top: 2rem; }
|
||||||
|
#chat article { margin: 0; padding: 0.75rem; }
|
||||||
|
#chat article audio { width: 100%; margin-bottom: 0.5rem; }
|
||||||
|
#chat article p { margin: 0; }
|
||||||
|
.input-area { padding: 0.5rem 0; display: flex; justify-content: center; }
|
||||||
|
#mic-btn {
|
||||||
|
width: 64px; height: 64px; border-radius: 50%; font-size: 1.8rem;
|
||||||
|
display: flex; align-items: center; justify-content: center;
|
||||||
|
cursor: pointer; border: none; padding: 0; min-height: 48px;
|
||||||
|
}
|
||||||
|
#mic-btn.recording { background: var(--pico-del-color); animation: pulse 1s infinite; }
|
||||||
|
@keyframes pulse { 0%, 100% { transform: scale(1); } 50% { transform: scale(1.1); } }
|
||||||
|
.spinner { display: inline-block; width: 1em; height: 1em; border: 2px solid currentColor; border-right-color: transparent; border-radius: 50%; animation: spin 0.6s linear infinite; vertical-align: middle; }
|
||||||
|
@keyframes spin { to { transform: rotate(360deg); } }
|
||||||
|
.error { color: var(--pico-del-color); }
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<main>
|
||||||
|
<h1>Transcribator</h1>
|
||||||
|
<div id="chat"></div>
|
||||||
|
<div class="input-area">
|
||||||
|
<button id="mic-btn" aria-label="Record">🎤</button>
|
||||||
|
</div>
|
||||||
|
</main>
|
||||||
|
<script>
|
||||||
|
const chat = document.getElementById('chat');
|
||||||
|
const micBtn = document.getElementById('mic-btn');
|
||||||
|
let mediaRecorder = null;
|
||||||
|
let chunks = [];
|
||||||
|
|
||||||
|
function getMimeType() {
|
||||||
|
if (MediaRecorder.isTypeSupported('audio/webm;codecs=opus')) return 'audio/webm';
|
||||||
|
if (MediaRecorder.isTypeSupported('audio/mp4')) return 'audio/mp4';
|
||||||
|
return '';
|
||||||
|
}
|
||||||
|
|
||||||
|
function getExtension(mime) {
|
||||||
|
if (mime.includes('webm')) return 'webm';
|
||||||
|
if (mime.includes('mp4')) return 'm4a';
|
||||||
|
return 'bin';
|
||||||
|
}
|
||||||
|
|
||||||
|
function addMessage(audioBlob) {
|
||||||
|
const article = document.createElement('article');
|
||||||
|
const audio = document.createElement('audio');
|
||||||
|
audio.controls = true;
|
||||||
|
audio.src = URL.createObjectURL(audioBlob);
|
||||||
|
const text = document.createElement('p');
|
||||||
|
text.innerHTML = '<span class="spinner"></span> Transcribing...';
|
||||||
|
article.append(audio, text);
|
||||||
|
chat.append(article);
|
||||||
|
chat.scrollTop = chat.scrollHeight;
|
||||||
|
return text;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function sendAudio(blob, textEl) {
|
||||||
|
const mime = blob.type || 'audio/webm';
|
||||||
|
const ext = getExtension(mime);
|
||||||
|
const form = new FormData();
|
||||||
|
form.append('audio', blob, `recording.${ext}`);
|
||||||
|
try {
|
||||||
|
const res = await fetch('/api/transcribe', { method: 'POST', body: form });
|
||||||
|
if (!res.ok) {
|
||||||
|
const err = await res.text();
|
||||||
|
textEl.innerHTML = `<span class="error">Error: ${err}</span>`;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const data = await res.json();
|
||||||
|
textEl.textContent = data.text || '(empty)';
|
||||||
|
} catch (e) {
|
||||||
|
textEl.innerHTML = `<span class="error">Error: ${e.message}</span>`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function toggleRecording() {
|
||||||
|
if (mediaRecorder && mediaRecorder.state === 'recording') {
|
||||||
|
mediaRecorder.stop();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const stream = await navigator.mediaDevices.getUserMedia({ audio: true });
|
||||||
|
const mime = getMimeType();
|
||||||
|
mediaRecorder = new MediaRecorder(stream, mime ? { mimeType: mime } : {});
|
||||||
|
chunks = [];
|
||||||
|
mediaRecorder.ondataavailable = e => { if (e.data.size > 0) chunks.push(e.data); };
|
||||||
|
mediaRecorder.onstop = () => {
|
||||||
|
stream.getTracks().forEach(t => t.stop());
|
||||||
|
micBtn.classList.remove('recording');
|
||||||
|
const blob = new Blob(chunks, { type: mediaRecorder.mimeType });
|
||||||
|
const textEl = addMessage(blob);
|
||||||
|
sendAudio(blob, textEl);
|
||||||
|
};
|
||||||
|
mediaRecorder.start();
|
||||||
|
micBtn.classList.add('recording');
|
||||||
|
} catch (e) {
|
||||||
|
alert('Microphone access denied: ' + e.message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
micBtn.addEventListener('click', toggleRecording);
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Reference in New Issue
Block a user