feat: add voice recognition

This commit is contained in:
qixinbo
2026-03-28 20:00:48 +08:00
parent 27270063f7
commit c983829392
7 changed files with 695 additions and 70 deletions
+1
View File
@@ -1,3 +1,4 @@
.vscode
nanobot-0.1.4.post4 nanobot-0.1.4.post4
data data
_research _research
+10
View File
@@ -761,6 +761,16 @@
"@noble/ciphers": "^1.0.0" "@noble/ciphers": "^1.0.0"
} }
}, },
"node_modules/@emnapi/runtime": {
"version": "1.9.1",
"resolved": "https://registry.npmmirror.com/@emnapi/runtime/-/runtime-1.9.1.tgz",
"integrity": "sha512-VYi5+ZVLhpgK4hQ0TAjiQiZ6ol0oe4mBx7mVv7IflsiEp0OWoVsp/+f9Vc1hOhE0TtkORVrI1GvzyreqpgWtkA==",
"license": "MIT",
"optional": true,
"dependencies": {
"tslib": "^2.4.0"
}
},
"node_modules/@esbuild/aix-ppc64": { "node_modules/@esbuild/aix-ppc64": {
"version": "0.21.5", "version": "0.21.5",
"resolved": "https://registry.npmmirror.com/@esbuild/aix-ppc64/-/aix-ppc64-0.21.5.tgz", "resolved": "https://registry.npmmirror.com/@esbuild/aix-ppc64/-/aix-ppc64-0.21.5.tgz",
+360 -70
View File
@@ -1,6 +1,6 @@
import { useState, useRef, useEffect } from "react"; import { useState, useRef, useEffect } from "react";
import { ScrollArea } from "@/components/ui/scroll-area"; import { ScrollArea } from "@/components/ui/scroll-area";
import { User, Loader2, ArrowUp, ChevronDown, Check, Square, Plus, Database, Wand2, Zap, CheckCircle2, Table, XCircle, Settings, ExternalLink, FileText, Download, Eye, Copy } from "lucide-react"; import { User, Loader2, ArrowUp, ChevronDown, Check, Square, Plus, Database, Wand2, Zap, CheckCircle2, Table, XCircle, Settings, ExternalLink, FileText, Download, Eye, Copy, Mic, X } from "lucide-react";
import { api } from "@/lib/api"; import { api } from "@/lib/api";
import { type ChartSpec } from "@/store/visualizationStore"; import { type ChartSpec } from "@/store/visualizationStore";
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover"; import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
@@ -14,7 +14,10 @@ import { useTranslation } from "react-i18next";
import { InlineVisualizationCard } from "./InlineVisualizationCard"; import { InlineVisualizationCard } from "./InlineVisualizationCard";
import { useProjectStore } from "@/store/projectStore"; import { useProjectStore } from "@/store/projectStore";
import { SlashCommandMenu } from "./SlashCommandMenu"; import { SlashCommandMenu } from "./SlashCommandMenu";
import { Dialog, DialogContent, DialogHeader, DialogTitle } from "@/components/ui/dialog"; import { Dialog, DialogContent, DialogHeader, DialogTitle, DialogFooter } from "@/components/ui/dialog";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Label } from "@/components/ui/label";
interface Message { interface Message {
id: string; id: string;
@@ -302,6 +305,145 @@ export function ChatInterface() {
const [, setIsUploading] = useState(false); const [, setIsUploading] = useState(false);
const fileInputRef = useRef<HTMLInputElement>(null); const fileInputRef = useRef<HTMLInputElement>(null);
// Speech Recognition State
const [isRecording, setIsRecording] = useState(false);
const [isTranscribing, setIsTranscribing] = useState(false);
const [recordingLevel, setRecordingLevel] = useState(0);
const mediaRecorderRef = useRef<MediaRecorder | null>(null);
const audioChunksRef = useRef<Blob[]>([]);
const shouldTranscribeRef = useRef(true);
const audioContextRef = useRef<AudioContext | null>(null);
const audioAnimationRef = useRef<number | null>(null);
// Local storage for whisper URL
const [whisperUrl, setWhisperUrl] = useState(() => localStorage.getItem("whisper_url") || "http://localhost:8001");
const [isVoiceSettingsOpen, setIsVoiceSettingsOpen] = useState(false);
const handleSaveWhisperUrl = (url: string) => {
setWhisperUrl(url);
localStorage.setItem("whisper_url", url);
setIsVoiceSettingsOpen(false);
};
const stopAudioMeter = () => {
if (audioAnimationRef.current) {
cancelAnimationFrame(audioAnimationRef.current);
audioAnimationRef.current = null;
}
if (audioContextRef.current) {
void audioContextRef.current.close();
audioContextRef.current = null;
}
setRecordingLevel(0);
};
const startAudioMeter = (stream: MediaStream) => {
const audioContext = new AudioContext();
const source = audioContext.createMediaStreamSource(stream);
const analyser = audioContext.createAnalyser();
analyser.fftSize = 1024;
source.connect(analyser);
audioContextRef.current = audioContext;
const dataArray = new Uint8Array(analyser.frequencyBinCount);
const tick = () => {
analyser.getByteTimeDomainData(dataArray);
let sum = 0;
for (let i = 0; i < dataArray.length; i += 1) {
const normalized = (dataArray[i] - 128) / 128;
sum += normalized * normalized;
}
const rms = Math.sqrt(sum / dataArray.length);
const level = Math.min(1, rms * 7);
setRecordingLevel(level);
audioAnimationRef.current = requestAnimationFrame(tick);
};
tick();
};
const startRecording = async () => {
try {
const stream = await navigator.mediaDevices.getUserMedia({ audio: true });
const mediaRecorder = new MediaRecorder(stream);
mediaRecorderRef.current = mediaRecorder;
audioChunksRef.current = [];
shouldTranscribeRef.current = true;
startAudioMeter(stream);
mediaRecorder.ondataavailable = (e) => {
if (e.data.size > 0) {
audioChunksRef.current.push(e.data);
}
};
mediaRecorder.onstop = async () => {
stopAudioMeter();
if (!shouldTranscribeRef.current) {
shouldTranscribeRef.current = true;
return;
}
setIsTranscribing(true);
try {
const audioBlob = new Blob(audioChunksRef.current, { type: 'audio/webm' });
const formData = new FormData();
formData.append("file", audioBlob, "audio.webm");
const baseUrl = whisperUrl || "http://localhost:8001";
const response = await fetch(`${baseUrl.replace(/\/$/, '')}/transcribe`, {
method: "POST",
body: formData,
});
if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`);
}
const output = await response.json();
if (output && output.text) {
setInput((prev) => prev + (prev ? " " : "") + output.text.trim());
}
} catch (err) {
console.error("Transcription error:", err);
} finally {
setIsTranscribing(false);
}
};
mediaRecorder.start();
setIsRecording(true);
} catch (err) {
console.error("Microphone access denied:", err);
}
};
const stopRecording = () => {
if (mediaRecorderRef.current && isRecording) {
mediaRecorderRef.current.stop();
setIsRecording(false);
mediaRecorderRef.current.stream.getTracks().forEach(track => track.stop());
}
};
const confirmRecording = () => {
shouldTranscribeRef.current = true;
stopRecording();
};
const cancelRecording = () => {
shouldTranscribeRef.current = false;
stopRecording();
};
useEffect(() => {
return () => {
stopAudioMeter();
if (mediaRecorderRef.current) {
mediaRecorderRef.current.stream.getTracks().forEach(track => track.stop());
}
};
}, []);
useEffect(() => { useEffect(() => {
fetchModels(); fetchModels();
}, []); }, []);
@@ -1071,41 +1213,101 @@ export function ChatInterface() {
</Popover> </Popover>
</div> </div>
<input {isRecording ? (
type="text" <>
value={input} <div className="flex-1 px-3">
onChange={handleInputChange} <div className="relative h-10 flex items-center">
onKeyDown={handleInputKeyDown} <div className="absolute inset-x-0 top-1/2 -translate-y-1/2 border-t border-dashed border-muted-foreground/40" />
placeholder={t('askAnything')} <div className="ml-auto flex items-center gap-[3px] pr-2">
className="flex-1 bg-transparent border-none focus:ring-0 text-lg px-3 py-2 text-foreground placeholder:text-muted-foreground/50 outline-none" {Array.from({ length: 30 }).map((_, idx) => {
disabled={isLoading} const dynamic = Math.abs(Math.sin(Date.now() / 180 + idx * 0.85));
/> const height = Math.max(4, Math.round((4 + dynamic * 18) * (0.45 + recordingLevel)));
<SlashCommandMenu return (
isOpen={slashQuery !== null} <span
skills={filteredSlashSkills} key={`recording-wave-empty-${idx}`}
selectedIndex={slashIndex} className="w-[3px] rounded-full bg-foreground/90 transition-all duration-75"
onSelect={handleSelectSlashSkill} style={{ height: `${height}px` }}
onClose={() => setSlashQuery(null)} />
/> );
})}
</div>
</div>
</div>
<div className="flex items-center gap-1">
<button
onClick={cancelRecording}
className="flex items-center justify-center h-10 w-10 rounded-full text-foreground hover:bg-muted transition-colors"
title={t('cancel', '取消')}
>
<X className="h-5 w-5" />
</button>
<button
onClick={confirmRecording}
className="flex items-center justify-center h-10 w-10 rounded-full text-foreground hover:bg-muted transition-colors"
title={t('confirm', '确认')}
>
<Check className="h-5 w-5" />
</button>
</div>
</>
) : (
<>
<input
type="text"
value={input}
onChange={handleInputChange}
onKeyDown={handleInputKeyDown}
placeholder={isTranscribing ? t('transcribing', '正在识别...') : t('askAnything')}
className="flex-1 bg-transparent border-none focus:ring-0 text-lg px-3 py-2 text-foreground placeholder:text-muted-foreground/50 outline-none"
disabled={isLoading || isTranscribing}
/>
<SlashCommandMenu
isOpen={slashQuery !== null}
skills={filteredSlashSkills}
selectedIndex={slashIndex}
onSelect={handleSelectSlashSkill}
onClose={() => setSlashQuery(null)}
/>
<div className="flex items-center gap-1"> <div className="flex items-center gap-1">
<button <button
onClick={handleSend} onClick={startRecording}
disabled={isLoading || !input.trim()} disabled={isLoading || isTranscribing}
className={cn( className="flex items-center justify-center h-10 w-10 rounded-full transition-all duration-200 bg-transparent text-muted-foreground hover:bg-muted"
"flex items-center justify-center h-10 w-10 rounded-full transition-all duration-200", title={t('voiceInput', '语音输入')}
(input.trim() || attachedFile || activeDataFile) && !isLoading >
? "bg-primary text-primary-foreground hover:bg-primary/90 shadow-sm" {isTranscribing ? (
: "bg-muted text-muted-foreground/50" <Loader2 className="h-5 w-5 animate-spin text-primary" />
)} ) : (
> <Mic className="h-5 w-5" />
{isLoading ? ( )}
<Loader2 className="h-5 w-5 animate-spin" /> </button>
) : ( <button
<ArrowUp className="h-6 w-6" /> onClick={() => setIsVoiceSettingsOpen(true)}
)} className="flex items-center justify-center h-10 w-10 rounded-full bg-transparent text-muted-foreground hover:bg-muted transition-colors"
</button> title={t('voiceSettings', '语音输入配置')}
</div> >
<Settings className="h-4 w-4" />
</button>
<button
onClick={handleSend}
disabled={isLoading || !input.trim()}
className={cn(
"flex items-center justify-center h-10 w-10 rounded-full transition-all duration-200",
(input.trim() || attachedFile || activeDataFile) && !isLoading
? "bg-primary text-primary-foreground hover:bg-primary/90 shadow-sm"
: "bg-muted text-muted-foreground/50"
)}
>
{isLoading ? (
<Loader2 className="h-5 w-5 animate-spin" />
) : (
<ArrowUp className="h-6 w-6" />
)}
</button>
</div>
</>
)}
</div> </div>
</div> </div>
<div className="mt-4 flex flex-wrap justify-center gap-2"> <div className="mt-4 flex flex-wrap justify-center gap-2">
@@ -1453,41 +1655,101 @@ export function ChatInterface() {
</Popover> </Popover>
</div> </div>
<input {isRecording ? (
type="text" <>
value={input} <div className="flex-1 px-3">
onChange={handleInputChange} <div className="relative h-10 flex items-center">
onKeyDown={handleInputKeyDown} <div className="absolute inset-x-0 top-1/2 -translate-y-1/2 border-t border-dashed border-muted-foreground/40" />
placeholder={t('askAnything')} <div className="ml-auto flex items-center gap-[3px] pr-2">
className="flex-1 bg-transparent border-none focus:ring-0 text-lg px-3 py-2 text-foreground placeholder:text-muted-foreground/50 outline-none" {Array.from({ length: 30 }).map((_, idx) => {
disabled={isLoading} const dynamic = Math.abs(Math.sin(Date.now() / 180 + idx * 0.85));
/> const height = Math.max(4, Math.round((4 + dynamic * 18) * (0.45 + recordingLevel)));
<SlashCommandMenu return (
isOpen={slashQuery !== null} <span
skills={filteredSlashSkills} key={`recording-wave-chat-${idx}`}
selectedIndex={slashIndex} className="w-[3px] rounded-full bg-foreground/90 transition-all duration-75"
onSelect={handleSelectSlashSkill} style={{ height: `${height}px` }}
onClose={() => setSlashQuery(null)} />
/> );
})}
</div>
</div>
</div>
<div className="flex items-center gap-1">
<button
onClick={cancelRecording}
className="flex items-center justify-center h-10 w-10 rounded-full text-foreground hover:bg-muted transition-colors"
title={t('cancel', '取消')}
>
<X className="h-5 w-5" />
</button>
<button
onClick={confirmRecording}
className="flex items-center justify-center h-10 w-10 rounded-full text-foreground hover:bg-muted transition-colors"
title={t('confirm', '确认')}
>
<Check className="h-5 w-5" />
</button>
</div>
</>
) : (
<>
<input
type="text"
value={input}
onChange={handleInputChange}
onKeyDown={handleInputKeyDown}
placeholder={isTranscribing ? t('transcribing', '正在识别...') : t('askAnything')}
className="flex-1 bg-transparent border-none focus:ring-0 text-lg px-3 py-2 text-foreground placeholder:text-muted-foreground/50 outline-none"
disabled={isLoading || isTranscribing}
/>
<SlashCommandMenu
isOpen={slashQuery !== null}
skills={filteredSlashSkills}
selectedIndex={slashIndex}
onSelect={handleSelectSlashSkill}
onClose={() => setSlashQuery(null)}
/>
<div className="flex items-center gap-1"> <div className="flex items-center gap-1">
<button <button
onClick={isLoading ? handleForceStop : handleSend} onClick={startRecording}
disabled={isLoading ? false : !input.trim()} disabled={isLoading || isTranscribing}
className={cn( className="flex items-center justify-center h-10 w-10 rounded-full transition-all duration-200 bg-transparent text-muted-foreground hover:bg-muted"
"flex items-center justify-center h-10 w-10 rounded-full transition-all duration-200", title={t('voiceInput', '语音输入')}
(input.trim() || isLoading) >
? (isLoading ? "bg-red-600 text-primary-foreground hover:bg-red-700" : "bg-primary text-primary-foreground hover:bg-primary/90") {isTranscribing ? (
: "bg-muted text-muted-foreground/50" <Loader2 className="h-5 w-5 animate-spin text-primary" />
)} ) : (
> <Mic className="h-5 w-5" />
{isLoading ? ( )}
<Square className="h-4 w-4" /> </button>
) : ( <button
<ArrowUp className="h-6 w-6" /> onClick={() => setIsVoiceSettingsOpen(true)}
)} className="flex items-center justify-center h-10 w-10 rounded-full bg-transparent text-muted-foreground hover:bg-muted transition-colors"
</button> title={t('voiceSettings', '语音输入配置')}
</div> >
<Settings className="h-4 w-4" />
</button>
<button
onClick={isLoading ? handleForceStop : handleSend}
disabled={isLoading ? false : !input.trim()}
className={cn(
"flex items-center justify-center h-10 w-10 rounded-full transition-all duration-200",
(input.trim() || isLoading)
? (isLoading ? "bg-red-600 text-primary-foreground hover:bg-red-700" : "bg-primary text-primary-foreground hover:bg-primary/90")
: "bg-muted text-muted-foreground/50"
)}
>
{isLoading ? (
<Square className="h-4 w-4" />
) : (
<ArrowUp className="h-6 w-6" />
)}
</button>
</div>
</>
)}
</div> </div>
</div> </div>
<div className="mt-2 flex justify-center"> <div className="mt-2 flex justify-center">
@@ -1534,6 +1796,34 @@ export function ChatInterface() {
</div> </div>
</DialogContent> </DialogContent>
</Dialog> </Dialog>
<Dialog open={isVoiceSettingsOpen} onOpenChange={setIsVoiceSettingsOpen}>
<DialogContent className="sm:max-w-[425px]">
<DialogHeader>
<DialogTitle>{t('voiceSettings', '语音输入配置')}</DialogTitle>
</DialogHeader>
<div className="grid gap-4 py-4">
<div className="grid grid-cols-4 items-center gap-4">
<Label htmlFor="whisperUrl" className="text-right">
{t('serviceUrl', '服务地址')}
</Label>
<Input
id="whisperUrl"
value={whisperUrl}
onChange={(e) => setWhisperUrl(e.target.value)}
className="col-span-3"
placeholder="http://localhost:8001"
/>
</div>
<p className="text-xs text-muted-foreground px-1">
Whisper http://localhost:8001
</p>
</div>
<DialogFooter>
<Button variant="outline" onClick={() => setIsVoiceSettingsOpen(false)}>{t('cancel', '取消')}</Button>
<Button onClick={() => handleSaveWhisperUrl(whisperUrl)}>{t('save', '保存')}</Button>
</DialogFooter>
</DialogContent>
</Dialog>
</div> </div>
); );
} }
+210
View File
@@ -0,0 +1,210 @@
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[codz]
*$py.class
data
*.db
# C extensions
*.so
# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST
# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec
# Installer logs
pip-log.txt
pip-delete-this-directory.txt
# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py.cover
.hypothesis/
.pytest_cache/
cover/
# Translations
*.mo
*.pot
# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal
# Flask stuff:
instance/
.webassets-cache
# Scrapy stuff:
.scrapy
# Sphinx documentation
docs/_build/
# PyBuilder
.pybuilder/
target/
# Jupyter Notebook
.ipynb_checkpoints
# IPython
profile_default/
ipython_config.py
# pyenv
# For a library or package, you might want to ignore these files since the code is
# intended to run in multiple environments; otherwise, check them in:
# .python-version
# pipenv
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
# However, in case of collaboration, if having platform-specific dependencies or dependencies
# having no cross-platform support, pipenv may install dependencies that don't work, or not
# install all needed dependencies.
#Pipfile.lock
# UV
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
#uv.lock
# poetry
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
#poetry.lock
#poetry.toml
# pdm
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
# pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
# https://pdm-project.org/en/latest/usage/project/#working-with-version-control
#pdm.lock
#pdm.toml
.pdm-python
.pdm-build/
# pixi
# Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
#pixi.lock
# Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
# in the .venv directory. It is recommended not to include this directory in version control.
.pixi
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
__pypackages__/
# Celery stuff
celerybeat-schedule
celerybeat.pid
# SageMath parsed files
*.sage.py
# Environments
.env
.envrc
.venv
env/
venv/
ENV/
env.bak/
venv.bak/
# Spyder project settings
.spyderproject
.spyproject
# Rope project settings
.ropeproject
# mkdocs documentation
/site
# mypy
.mypy_cache/
.dmypy.json
dmypy.json
# Pyre type checker
.pyre/
# pytype static type analyzer
.pytype/
# Cython debug symbols
cython_debug/
# PyCharm
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
# and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/
# Abstra
# Abstra is an AI-powered process automation framework.
# Ignore directories containing user credentials, local state, and settings.
# Learn more at https://abstra.io/docs
.abstra/
# Visual Studio Code
# Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore
# that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
# and can be added to the global gitignore or merged into this file. However, if you prefer,
# you could uncomment the following to ignore the entire vscode folder
# .vscode/
# Ruff stuff:
.ruff_cache/
# PyPI configuration file
.pypirc
# Cursor
# Cursor is an AI-powered code editor. `.cursorignore` specifies files/directories to
# exclude from AI features like autocomplete and code analysis. Recommended for sensitive data
# refer to https://docs.cursor.com/context/ignore-files
.cursorignore
.cursorindexingignore
# Marimo
marimo/_static/
marimo/_lsp/
__marimo__/
+39
View File
@@ -0,0 +1,39 @@
# Whisper Transcription Service
This is a standalone HTTP service for transcribing audio files using the OpenAI Whisper model.
## Prerequisites
Make sure you have Python 3.9+ and `ffmpeg` installed on your system.
To install `ffmpeg` on macOS:
```bash
brew install ffmpeg
```
## Setup & Run
1. Create a virtual environment and install dependencies:
```bash
cd whisper
python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
```
2. Start the server:
```bash
python main.py
```
Or run with uvicorn directly:
```bash
uvicorn main:app --host 0.0.0.0 --port 8001 --reload
```
The service will run on `http://localhost:8001`.
## API Endpoint
- `POST /transcribe`
- Body: `multipart/form-data` with a `file` field containing the audio blob.
- Returns: `{"text": "transcribed text..."}`
+71
View File
@@ -0,0 +1,71 @@
import os
import shutil
import ssl
import tempfile
import traceback
from fastapi import FastAPI, File, UploadFile
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
import whisper
import imageio_ffmpeg
# Ensure whisper can execute "ffmpeg" command
_ffmpeg_exe = imageio_ffmpeg.get_ffmpeg_exe()
_ffmpeg_bin_dir = os.path.join(tempfile.gettempdir(), "dataclaw_ffmpeg_bin")
_ffmpeg_link = os.path.join(_ffmpeg_bin_dir, "ffmpeg")
os.makedirs(_ffmpeg_bin_dir, exist_ok=True)
if not os.path.exists(_ffmpeg_link):
try:
os.symlink(_ffmpeg_exe, _ffmpeg_link)
except OSError:
shutil.copy2(_ffmpeg_exe, _ffmpeg_link)
os.chmod(_ffmpeg_link, 0o755)
os.environ["PATH"] = _ffmpeg_bin_dir + os.pathsep + os.environ.get("PATH", "")
# Disable SSL verification temporarily to fix UNEXPECTED_EOF_WHILE_READING error during model download
ssl._create_default_https_context = ssl._create_unverified_context
app = FastAPI(title="Whisper Transcription Service")
# Allow CORS for frontend
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # Allows all origins
allow_credentials=True,
allow_methods=["*"], # Allows all methods
allow_headers=["*"], # Allows all headers
)
# Load the whisper model globally. "small" is a good balance between speed and accuracy.
print("Loading Whisper model (small)... This may take a moment.")
model = whisper.load_model("small")
print("Model loaded successfully.")
@app.post("/transcribe")
async def transcribe_audio(file: UploadFile = File(...)):
# Save the uploaded file to a temporary file
with tempfile.NamedTemporaryFile(delete=False, suffix=".webm") as tmp:
content = await file.read()
tmp.write(content)
tmp_path = tmp.name
try:
# Convert webm to wav since Whisper's internal ffmpeg dependency can be problematic
# We will use an alternative approach or just pass the webm if it works natively
# Transcribe using whisper
# Forcing language to Chinese for better accuracy on Chinese speech
result = model.transcribe(tmp_path, language="zh", task="transcribe")
return {"text": result.get("text", "")}
except Exception as e:
print(f"Error during transcription: {e}")
print(traceback.format_exc())
return JSONResponse(status_code=500, content={"error": str(e)})
finally:
# Clean up the temporary file
if os.path.exists(tmp_path):
os.remove(tmp_path)
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8001)
+4
View File
@@ -0,0 +1,4 @@
fastapi
uvicorn
python-multipart
openai-whisper