Spaces:
Sleeping
Sleeping
Commit
·
755dbb1
1
Parent(s):
3b598e6
Improved comments and docstrings
Browse files
app.py
CHANGED
|
@@ -11,31 +11,60 @@ from utils.params import default_audio_params
|
|
| 11 |
|
| 12 |
|
| 13 |
def initialize_services():
|
| 14 |
-
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
config = Config()
|
| 16 |
llm = LLMManager(config, prompts)
|
| 17 |
tts = TTSManager(config)
|
| 18 |
stt = STTManager(config)
|
|
|
|
|
|
|
| 19 |
default_audio_params["streaming"] = stt.streaming
|
|
|
|
|
|
|
| 20 |
if os.getenv("SILENT", False):
|
| 21 |
tts.read_last_message = lambda x: None
|
|
|
|
| 22 |
return config, llm, tts, stt
|
| 23 |
|
| 24 |
|
| 25 |
def create_interface(llm, tts, stt, audio_params):
|
| 26 |
-
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
with gr.Blocks(title="AI Interviewer", theme=gr.themes.Default()) as demo:
|
|
|
|
| 28 |
audio_output = gr.Audio(label="Play audio", autoplay=True, visible=os.environ.get("DEBUG", False), streaming=tts.streaming)
|
|
|
|
|
|
|
| 29 |
get_problem_solving_ui(llm, tts, stt, audio_params, audio_output).render()
|
| 30 |
get_instructions_ui(llm, tts, stt, audio_params).render()
|
|
|
|
| 31 |
return demo
|
| 32 |
|
| 33 |
|
| 34 |
def main():
|
| 35 |
-
"""
|
|
|
|
|
|
|
| 36 |
config, llm, tts, stt = initialize_services()
|
| 37 |
demo = create_interface(llm, tts, stt, default_audio_params)
|
| 38 |
demo.config["dependencies"][0]["show_progress"] = "hidden"
|
|
|
|
|
|
|
| 39 |
demo.launch(show_api=False)
|
| 40 |
|
| 41 |
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
def initialize_services():
|
| 14 |
+
"""
|
| 15 |
+
Initialize configuration, LLM, TTS, and STT services.
|
| 16 |
+
|
| 17 |
+
Returns:
|
| 18 |
+
tuple: Containing Config, LLMManager, TTSManager, and STTManager instances.
|
| 19 |
+
"""
|
| 20 |
config = Config()
|
| 21 |
llm = LLMManager(config, prompts)
|
| 22 |
tts = TTSManager(config)
|
| 23 |
stt = STTManager(config)
|
| 24 |
+
|
| 25 |
+
# Update default audio parameters with STT streaming setting
|
| 26 |
default_audio_params["streaming"] = stt.streaming
|
| 27 |
+
|
| 28 |
+
# Disable TTS in silent mode
|
| 29 |
if os.getenv("SILENT", False):
|
| 30 |
tts.read_last_message = lambda x: None
|
| 31 |
+
|
| 32 |
return config, llm, tts, stt
|
| 33 |
|
| 34 |
|
| 35 |
def create_interface(llm, tts, stt, audio_params):
|
| 36 |
+
"""
|
| 37 |
+
Create and configure the Gradio interface.
|
| 38 |
+
|
| 39 |
+
Args:
|
| 40 |
+
llm (LLMManager): Language model manager instance.
|
| 41 |
+
tts (TTSManager): Text-to-speech manager instance.
|
| 42 |
+
stt (STTManager): Speech-to-text manager instance.
|
| 43 |
+
audio_params (dict): Audio parameters for the interface.
|
| 44 |
+
|
| 45 |
+
Returns:
|
| 46 |
+
gr.Blocks: Configured Gradio interface.
|
| 47 |
+
"""
|
| 48 |
with gr.Blocks(title="AI Interviewer", theme=gr.themes.Default()) as demo:
|
| 49 |
+
# Create audio output component (visible only in debug mode)
|
| 50 |
audio_output = gr.Audio(label="Play audio", autoplay=True, visible=os.environ.get("DEBUG", False), streaming=tts.streaming)
|
| 51 |
+
|
| 52 |
+
# Render problem-solving and instructions UI components
|
| 53 |
get_problem_solving_ui(llm, tts, stt, audio_params, audio_output).render()
|
| 54 |
get_instructions_ui(llm, tts, stt, audio_params).render()
|
| 55 |
+
|
| 56 |
return demo
|
| 57 |
|
| 58 |
|
| 59 |
def main():
|
| 60 |
+
"""
|
| 61 |
+
Main function to initialize services and launch the Gradio interface.
|
| 62 |
+
"""
|
| 63 |
config, llm, tts, stt = initialize_services()
|
| 64 |
demo = create_interface(llm, tts, stt, default_audio_params)
|
| 65 |
demo.config["dependencies"][0]["show_progress"] = "hidden"
|
| 66 |
+
|
| 67 |
+
# Launch the Gradio interface
|
| 68 |
demo.launch(show_api=False)
|
| 69 |
|
| 70 |
|