pwenker commited on
Commit
0c78e45
·
1 Parent(s): 6c973ca

feat: Allow users to paste in their own openai api key

Browse files
src/pronunciation_trainer/app.py CHANGED
@@ -5,6 +5,7 @@ This script provides a simple web interface for the pronunciation trainer using
5
 
6
  from pathlib import Path
7
 
 
8
  import gradio as gr
9
  from phonemizer import phonemize
10
 
@@ -88,6 +89,15 @@ with gr.Blocks() as demo:
88
  learner_l2 = gr.Textbox(
89
  label="Language the learner aims to acquire (L2)", placeholder="English"
90
  )
 
 
 
 
 
 
 
 
 
91
  advanced_evaluate_btn = gr.Button("Evaluate", variant="primary")
92
  gr.Markdown("## Advanced Evaluation")
93
 
@@ -125,6 +135,7 @@ with gr.Blocks() as demo:
125
  learner_l2,
126
  learner_phoneme_transcription,
127
  teacher_phoneme_transcription,
 
128
  ],
129
  outputs=llm_evaluation,
130
  )
 
5
 
6
  from pathlib import Path
7
 
8
+ import os
9
  import gradio as gr
10
  from phonemizer import phonemize
11
 
 
89
  learner_l2 = gr.Textbox(
90
  label="Language the learner aims to acquire (L2)", placeholder="English"
91
  )
92
+ openai_api_key = gr.Textbox(
93
+ placeholder="Paste your OpenAI API key (sk-...)",
94
+ show_label=False,
95
+ value=os.getenv("OPENAI_API_KEY"),
96
+ label="openai_api_key",
97
+ lines=1,
98
+ type="password",
99
+ )
100
+
101
  advanced_evaluate_btn = gr.Button("Evaluate", variant="primary")
102
  gr.Markdown("## Advanced Evaluation")
103
 
 
135
  learner_l2,
136
  learner_phoneme_transcription,
137
  teacher_phoneme_transcription,
138
+ openai_api_key,
139
  ],
140
  outputs=llm_evaluation,
141
  )
src/pronunciation_trainer/evaluation.py CHANGED
@@ -73,9 +73,10 @@ def advanced_evaluation(
73
  learner_l2,
74
  learner_phoneme_transcription,
75
  teacher_phoneme_transcription,
 
76
  ) -> str:
77
  """Provide LLM-based feedback"""
78
- return create_llm_chain().invoke(
79
  {
80
  "learner_l1": learner_l1,
81
  "learner_l2": learner_l2,
 
73
  learner_l2,
74
  learner_phoneme_transcription,
75
  teacher_phoneme_transcription,
76
+ openai_api_key,
77
  ) -> str:
78
  """Provide LLM-based feedback"""
79
+ return create_llm_chain(openai_api_key=openai_api_key).invoke(
80
  {
81
  "learner_l1": learner_l1,
82
  "learner_l2": learner_l2,