chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,208 @@
|
||||
"""
|
||||
FunASR Voice Input — 语音输入法
|
||||
|
||||
安装:
|
||||
pip install funasr sounddevice numpy pyperclip openai
|
||||
|
||||
使用:
|
||||
1. 先启动 funasr-server: funasr-server --device cuda (或 cpu)
|
||||
2. 运行: python funasr_input.py
|
||||
3. 按住 空格键 说话，松开自动识别并输入到光标位置
|
||||
|
||||
配置:
|
||||
--server FunASR server 地址 (默认 http://localhost:8000/v1)
|
||||
--key 快捷键 (默认 ctrl+shift+space)
|
||||
--lang 语言 (默认 auto)
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
import os
|
||||
import tempfile
|
||||
import threading
|
||||
import time
|
||||
import io
|
||||
import wave
|
||||
import numpy as np
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
description="FunASR Voice Input - 语音输入法",
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
epilog="""
|
||||
使用方法:
|
||||
1. 启动 FunASR 服务: funasr-server --device cuda
|
||||
2. 运行本程序: python funasr_input.py
|
||||
3. 按 Ctrl+Shift+Space 开始录音，再按一次停止并输入文字
|
||||
|
||||
示例:
|
||||
python funasr_input.py # 默认配置
|
||||
python funasr_input.py --server http://gpu:8000/v1 # 远程服务器
|
||||
python funasr_input.py --model paraformer # 指定模型
|
||||
""",
|
||||
)
|
||||
parser.add_argument("--server", default="http://localhost:8000/v1", help="FunASR server URL")
|
||||
parser.add_argument("--model", default="sensevoice", help="ASR model")
|
||||
parser.add_argument("--lang", default="auto", help="Language hint")
|
||||
parser.add_argument("--rate", type=int, default=16000, help="Sample rate")
|
||||
parser.add_argument("--hotkey", default="ctrl+shift+space", help="Hotkey to toggle recording")
|
||||
args = parser.parse_args()
|
||||
|
||||
try:
|
||||
import sounddevice as sd
|
||||
except ImportError:
|
||||
print("Error: sounddevice required. Install: pip install sounddevice")
|
||||
sys.exit(1)
|
||||
try:
|
||||
import pyperclip
|
||||
except ImportError:
|
||||
print("Error: pyperclip required. Install: pip install pyperclip")
|
||||
sys.exit(1)
|
||||
try:
|
||||
from openai import OpenAI
|
||||
except ImportError:
|
||||
print("Error: openai required. Install: pip install openai")
|
||||
sys.exit(1)
|
||||
try:
|
||||
from pynput import keyboard
|
||||
except ImportError:
|
||||
print("Error: pynput required. Install: pip install pynput")
|
||||
sys.exit(1)
|
||||
|
||||
client = OpenAI(base_url=args.server, api_key="not-needed")
|
||||
|
||||
# State
|
||||
recording = False
|
||||
audio_frames = []
|
||||
lock = threading.Lock()
|
||||
|
||||
def start_recording():
|
||||
nonlocal recording, audio_frames
|
||||
with lock:
|
||||
if recording:
|
||||
return
|
||||
recording = True
|
||||
audio_frames = []
|
||||
print("🎤 录音中... (再按快捷键停止)")
|
||||
|
||||
def callback(indata, frames, time_info, status):
|
||||
if recording:
|
||||
audio_frames.append(indata.copy())
|
||||
|
||||
sd.default.samplerate = args.rate
|
||||
sd.default.channels = 1
|
||||
sd.default.dtype = 'int16'
|
||||
|
||||
stream = sd.InputStream(callback=callback)
|
||||
stream.start()
|
||||
|
||||
# Wait until recording stops
|
||||
while recording:
|
||||
time.sleep(0.05)
|
||||
stream.stop()
|
||||
stream.close()
|
||||
|
||||
def stop_and_transcribe():
|
||||
nonlocal recording
|
||||
with lock:
|
||||
if not recording:
|
||||
return
|
||||
recording = False
|
||||
|
||||
print("⏳ 识别中...")
|
||||
|
||||
if not audio_frames:
|
||||
print("❌ 没有录到音频")
|
||||
return
|
||||
|
||||
# Convert to WAV
|
||||
audio_data = np.concatenate(audio_frames, axis=0)
|
||||
wav_buffer = io.BytesIO()
|
||||
with wave.open(wav_buffer, 'wb') as wf:
|
||||
wf.setnchannels(1)
|
||||
wf.setsampwidth(2)
|
||||
wf.setframerate(args.rate)
|
||||
wf.writeframes(audio_data.tobytes())
|
||||
wav_buffer.seek(0)
|
||||
|
||||
# Send to FunASR server
|
||||
try:
|
||||
result = client.audio.transcriptions.create(
|
||||
model=args.model,
|
||||
file=("recording.wav", wav_buffer, "audio/wav"),
|
||||
)
|
||||
text = result.text.strip()
|
||||
if text:
|
||||
# Copy to clipboard and simulate paste
|
||||
pyperclip.copy(text)
|
||||
print(f"✅ {text}")
|
||||
print(" (已复制到剪贴板，Cmd+V/Ctrl+V 粘贴)")
|
||||
|
||||
# Try to auto-paste (macOS)
|
||||
if sys.platform == 'darwin':
|
||||
os.system('osascript -e \'tell application "System Events" to keystroke "v" using command down\'')
|
||||
elif sys.platform == 'linux':
|
||||
os.system('xdotool key ctrl+v 2>/dev/null')
|
||||
else:
|
||||
print("❌ 未识别到语音")
|
||||
except Exception as e:
|
||||
print(f"❌ 识别失败: {e}")
|
||||
print(f" 请确认 funasr-server 正在运行: funasr-server --device cuda")
|
||||
|
||||
# Hotkey handling
|
||||
print("=" * 50)
|
||||
print(" FunASR 语音输入法")
|
||||
print("=" * 50)
|
||||
print(f" 服务器: {args.server}")
|
||||
print(f" 模型: {args.model}")
|
||||
print(f" 快捷键: {args.hotkey}")
|
||||
print(f" 语言: {args.lang}")
|
||||
print("=" * 50)
|
||||
print(f"按 {args.hotkey} 开始/停止录音")
|
||||
print("按 Ctrl+C 退出")
|
||||
print()
|
||||
|
||||
# Parse hotkey
|
||||
hotkey_parts = args.hotkey.lower().split('+')
|
||||
modifiers = set()
|
||||
key_char = None
|
||||
for part in hotkey_parts:
|
||||
if part in ('ctrl', 'control'):
|
||||
modifiers.add(keyboard.Key.ctrl_l)
|
||||
elif part in ('shift',):
|
||||
modifiers.add(keyboard.Key.shift_l)
|
||||
elif part in ('alt', 'option'):
|
||||
modifiers.add(keyboard.Key.alt_l)
|
||||
elif part in ('cmd', 'command', 'super'):
|
||||
modifiers.add(keyboard.Key.cmd)
|
||||
elif part == 'space':
|
||||
key_char = keyboard.Key.space
|
||||
else:
|
||||
key_char = keyboard.KeyCode.from_char(part)
|
||||
|
||||
current_modifiers = set()
|
||||
record_thread = None
|
||||
|
||||
def on_press(key):
|
||||
nonlocal record_thread
|
||||
if key in modifiers or (hasattr(key, 'value') and key.value in [k.value for k in modifiers if hasattr(k, 'value')]):
|
||||
current_modifiers.add(key)
|
||||
if key == key_char and len(current_modifiers) >= len(modifiers):
|
||||
if not recording:
|
||||
record_thread = threading.Thread(target=start_recording, daemon=True)
|
||||
record_thread.start()
|
||||
else:
|
||||
stop_and_transcribe()
|
||||
|
||||
def on_release(key):
|
||||
current_modifiers.discard(key)
|
||||
|
||||
with keyboard.Listener(on_press=on_press, on_release=on_release) as listener:
|
||||
try:
|
||||
listener.join()
|
||||
except KeyboardInterrupt:
|
||||
print("\n👋 退出")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user