Spaces:
Sleeping
Sleeping
Upload 5 files
Browse files- .gitattributes +3 -1
- .gitignore +6 -0
- README.md +33 -21
- app.py +24 -72
- requirements.txt +1 -14
.gitattributes
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
-
*
|
|
|
|
|
|
|
| 2 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 1 |
+
* text=auto eol=lf
|
| 2 |
+
|
| 3 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 4 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 5 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 6 |
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
.gitignore
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
.venv/
|
| 2 |
+
__pycache__/
|
| 3 |
+
*.py[cod]
|
| 4 |
+
.env
|
| 5 |
+
analysis_output.txt
|
| 6 |
+
*.wav
|
README.md
CHANGED
|
@@ -1,42 +1,54 @@
|
|
| 1 |
---
|
| 2 |
title: GTROX
|
| 3 |
-
emoji:
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 6.5.1
|
| 8 |
app_file: app.py
|
| 9 |
pinned: true
|
| 10 |
-
hf_oauth: true
|
| 11 |
-
hf_oauth_scopes:
|
| 12 |
-
- inference-api
|
| 13 |
---
|
| 14 |
|
| 15 |
-
#
|
| 16 |
|
| 17 |
-
|
|
|
|
| 18 |
|
| 19 |
-
##
|
| 20 |
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
|
|
|
|
|
|
| 25 |
|
| 26 |
-
|
| 27 |
|
|
|
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
-
|
| 31 |
-
2. **Analysis:** It calculates technical metrics like WPM and silence ratios using numerical analysis.
|
| 32 |
-
3. **Transcription:** Uses `distil-whisper` for highly accurate, low-latency speech-to-text.
|
| 33 |
-
4. **Feedback:** The `SmolLM2` language model analyzes your performance and suggests improvements.
|
| 34 |
|
| 35 |
-
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
-
|
|
|
|
| 38 |
|
| 39 |
-
##
|
| 40 |
-
|
| 41 |
-
To run this application locally, ensure you have Python 3.10+ and the necessary dependencies installed:
|
| 42 |
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
title: GTROX
|
| 3 |
+
emoji: ๐๏ธ
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 6.5.1
|
| 8 |
app_file: app.py
|
| 9 |
pinned: true
|
|
|
|
|
|
|
|
|
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# GTROX Speech Coach
|
| 13 |
|
| 14 |
+
GTROX analyzes recorded speech and returns a transcript, measurable delivery
|
| 15 |
+
KPIs, and personalized coaching.
|
| 16 |
|
| 17 |
+
## Pipeline
|
| 18 |
|
| 19 |
+
1. The Gradio app accepts an uploaded file or microphone recording.
|
| 20 |
+
2. NVIDIA Parakeet runs on Modal and produces the transcript.
|
| 21 |
+
3. Local deterministic analysis calculates WPM, filler density, repetitions,
|
| 22 |
+
and a fluency score.
|
| 23 |
+
4. Qwen2.5-1.5B-Instruct runs on a Modal T4 and generates grounded coaching.
|
| 24 |
+
5. KPI-based coaching is returned automatically if the AI coach is unavailable.
|
| 25 |
|
| 26 |
+
The public transcription service limits uploads to 25 MB and five minutes.
|
| 27 |
|
| 28 |
+
## Project Structure
|
| 29 |
|
| 30 |
+
```text
|
| 31 |
+
app.py Gradio UI
|
| 32 |
+
gtrox/config.py Endpoint configuration
|
| 33 |
+
gtrox/clients.py Modal HTTP clients
|
| 34 |
+
gtrox/metrics.py Deterministic KPI calculations
|
| 35 |
+
gtrox/pipeline.py End-to-end orchestration
|
| 36 |
+
services/transcribe.py Modal Parakeet service
|
| 37 |
+
services/coach.py Modal Qwen coaching service
|
| 38 |
+
tests/test_metrics.py KPI unit tests
|
| 39 |
+
```
|
| 40 |
|
| 41 |
+
## Deploy Modal Services
|
|
|
|
|
|
|
|
|
|
| 42 |
|
| 43 |
+
```bash
|
| 44 |
+
modal deploy -m services.transcribe
|
| 45 |
+
modal deploy -m services.coach
|
| 46 |
+
```
|
| 47 |
|
| 48 |
+
The endpoint defaults are stored in `gtrox/config.py` and can be overridden
|
| 49 |
+
with the `TRANSCRIBE_URL` and `COACH_URL` environment variables.
|
| 50 |
|
| 51 |
+
## Current Limitations
|
|
|
|
|
|
|
| 52 |
|
| 53 |
+
Version 1 does not claim to measure pauses, tone, emotion, or pronunciation.
|
| 54 |
+
Those features require a reliable timestamp or voice-activity detection layer.
|
app.py
CHANGED
|
@@ -1,82 +1,34 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
import librosa
|
| 3 |
-
import numpy as np
|
| 4 |
-
import torch
|
| 5 |
-
import soundfile as sf
|
| 6 |
-
from transformers import pipeline
|
| 7 |
-
import os
|
| 8 |
|
| 9 |
-
|
| 10 |
-
# We set to 'cpu' as default for broader compatibility, but it will detect 'cuda' if available.
|
| 11 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 12 |
|
| 13 |
-
# Transcriber: Fast, efficient speech-to-text
|
| 14 |
-
transcriber = pipeline("automatic-speech-recognition",
|
| 15 |
-
model="distil-whisper/distil-small.en",
|
| 16 |
-
device=device,
|
| 17 |
-
return_timestamps=True
|
| 18 |
-
)
|
| 19 |
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
def analyze_speech(audio_path):
|
| 26 |
-
if audio_path is None:
|
| 27 |
-
return "Please upload an audio file."
|
| 28 |
-
|
| 29 |
-
# 1. Load Data
|
| 30 |
-
try:
|
| 31 |
-
data, sr = sf.read(audio_path)
|
| 32 |
-
if len(data.shape) > 1: data = data.mean(axis=1)
|
| 33 |
-
if sr != 16000:
|
| 34 |
-
data = librosa.resample(data, orig_sr=sr, target_sr=16000)
|
| 35 |
-
except Exception as e:
|
| 36 |
-
return f"Error loading audio: {str(e)}"
|
| 37 |
-
|
| 38 |
-
# 2. Transcription (Using optimized chunking)
|
| 39 |
-
# This prevents the "long-form" 30s crash
|
| 40 |
-
try:
|
| 41 |
-
# Pass the array and sample rate directly
|
| 42 |
-
result = transcriber({"raw": data, "sampling_rate": 16000})
|
| 43 |
-
text = result["text"]
|
| 44 |
-
except Exception as e:
|
| 45 |
-
return f"Transcription error: {str(e)}"
|
| 46 |
-
|
| 47 |
-
# 3. Metrics
|
| 48 |
-
duration = len(data) / 16000
|
| 49 |
-
wpm = (len(text.split()) / (duration / 60)) if duration > 0 else 0
|
| 50 |
-
silence_ratio = np.sum(np.abs(data) < 0.01) / len(data)
|
| 51 |
-
|
| 52 |
-
# 4. Feedback
|
| 53 |
-
prompt = (
|
| 54 |
-
f"You are a professional speech coach. The user spoke at {wpm:.1f} WPM with {silence_ratio:.1%} silence. "
|
| 55 |
-
f"Transcript: '{text}'. "
|
| 56 |
-
"Provide exactly 2 short, constructive tips to improve their delivery, focusing on pacing and clarity."
|
| 57 |
)
|
| 58 |
-
try:
|
| 59 |
-
# Simplified generation call
|
| 60 |
-
feedback = coach(prompt, max_new_tokens=100)[0]['generated_text']
|
| 61 |
-
except Exception as e:
|
| 62 |
-
feedback = "Could not generate feedback."
|
| 63 |
|
| 64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
|
| 67 |
-
# 2. UI Layout
|
| 68 |
-
with gr.Blocks() as demo:
|
| 69 |
-
gr.Markdown("# ๐๏ธ ImproveTalk: Speech Insight Coach")
|
| 70 |
-
gr.Markdown("Upload your `.wav` file to get instant feedback on your fluency.")
|
| 71 |
-
|
| 72 |
-
with gr.Row():
|
| 73 |
-
# audio_input = gr.Audio(type="filepath", label="Upload WAV Audio")
|
| 74 |
-
audio_input = gr.Audio(type="filepath", label="Record or Upload Audio",sources=["microphone", "upload"])
|
| 75 |
-
analyze_btn = gr.Button("Analyze My Speech", variant="primary")
|
| 76 |
-
|
| 77 |
-
results = gr.Textbox(label="Analysis Results", lines=10)
|
| 78 |
-
|
| 79 |
-
analyze_btn.click(analyze_speech, inputs=audio_input, outputs=results)
|
| 80 |
|
| 81 |
if __name__ == "__main__":
|
| 82 |
-
demo.launch()
|
|
|
|
| 1 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
+
from gtrox.pipeline import analyze_audio
|
|
|
|
|
|
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
+
with gr.Blocks(title="GTROX Speech Coach") as demo:
|
| 7 |
+
gr.Markdown("# GTROX Speech Coach")
|
| 8 |
+
gr.Markdown(
|
| 9 |
+
"GPU transcription by NVIDIA Parakeet, delivery KPIs, and personalized "
|
| 10 |
+
"coaching by Qwen2.5."
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
+
audio_input = gr.Audio(
|
| 14 |
+
type="filepath",
|
| 15 |
+
label="Record or Upload Audio",
|
| 16 |
+
sources=["microphone", "upload"],
|
| 17 |
+
)
|
| 18 |
+
analyze_btn = gr.Button("Analyze Speech", variant="primary")
|
| 19 |
+
transcript_box = gr.Textbox(label="Transcript", lines=8)
|
| 20 |
+
metrics_table = gr.Dataframe(
|
| 21 |
+
headers=["Metric", "Value"],
|
| 22 |
+
interactive=False,
|
| 23 |
+
)
|
| 24 |
+
feedback_box = gr.Markdown(label="Coach Feedback")
|
| 25 |
|
| 26 |
+
analyze_btn.click(
|
| 27 |
+
analyze_audio,
|
| 28 |
+
inputs=audio_input,
|
| 29 |
+
outputs=[transcript_box, metrics_table, feedback_box],
|
| 30 |
+
)
|
| 31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
if __name__ == "__main__":
|
| 34 |
+
demo.launch()
|
requirements.txt
CHANGED
|
@@ -1,16 +1,3 @@
|
|
| 1 |
-
# UI Framework
|
| 2 |
gradio>=5.0.0
|
| 3 |
-
|
| 4 |
-
# Core AI/ML Libraries
|
| 5 |
-
torch>=2.0.0
|
| 6 |
-
transformers>=4.40.0
|
| 7 |
-
accelerate>=0.30.0
|
| 8 |
-
|
| 9 |
-
# Audio Processing
|
| 10 |
-
librosa>=0.10.0
|
| 11 |
soundfile>=0.12.0
|
| 12 |
-
numpy>=1.26.0
|
| 13 |
-
|
| 14 |
-
# Inference Optimization (Crucial for Small Models)
|
| 15 |
-
bitsandbytes>=0.43.0
|
| 16 |
-
|
|
|
|
|
|
|
| 1 |
gradio>=5.0.0
|
| 2 |
+
requests>=2.32.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
soundfile>=0.12.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|