Ridealist commited on
Commit
629cc58
1 Parent(s): 8e32e83

fix: radio selection disconnect api call

Browse files
Files changed (1) hide show
  1. src/obs_eval_gradio.py +31 -27
src/obs_eval_gradio.py CHANGED
@@ -250,36 +250,40 @@ def call_gpt_audio(api_key, rubrics) -> str:
250
 
251
  full_text_audio = ""
252
 
253
- PROMPT_MESSAGES = [
254
- {
255
- "role": "system",
256
- "content": AUDIO_SYSTEM_PROMPT,
257
- },
258
- {
259
- "role": "user",
260
- "content": PromptTemplate.from_template(USER_PROMPT_TEMPLATE).format(rubrics=rubrics) + "\n\n<TEXT>\n" + transcript
261
- },
262
- ]
263
- params = {
264
- "model": "gpt-4",
265
- "messages": PROMPT_MESSAGES,
266
- "max_tokens": 1024,
267
- }
 
 
 
 
268
 
269
- try:
270
- result = openai.chat.completions.create(**params)
271
- full_text_audio = result.choices[0].message.content
272
- print(full_text_audio)
273
- except openai.OpenAIError as e:
274
- print(f"Failed to connect to OpenAI: {e}")
275
- pass
276
 
277
- if 'full_text_audio' not in global_dict:
278
- global_dict.setdefault('full_text_audio', full_text_audio)
279
- else:
280
- global_dict['full_text_audio'] = full_text_audio
281
 
282
- return full_text_audio
283
 
284
 
285
  def get_full_result():
 
250
 
251
  full_text_audio = ""
252
 
253
+ print(f"RUBRIC_AUDIO: {rubrics}")
254
+ if not rubrics:
255
+ return full_text_audio
256
+ else:
257
+ PROMPT_MESSAGES = [
258
+ {
259
+ "role": "system",
260
+ "content": AUDIO_SYSTEM_PROMPT,
261
+ },
262
+ {
263
+ "role": "user",
264
+ "content": PromptTemplate.from_template(USER_PROMPT_TEMPLATE).format(rubrics=rubrics) + "\n\n<TEXT>\n" + transcript
265
+ },
266
+ ]
267
+ params = {
268
+ "model": "gpt-4",
269
+ "messages": PROMPT_MESSAGES,
270
+ "max_tokens": 1024,
271
+ }
272
 
273
+ try:
274
+ result = openai.chat.completions.create(**params)
275
+ full_text_audio = result.choices[0].message.content
276
+ print(full_text_audio)
277
+ except openai.OpenAIError as e:
278
+ print(f"Failed to connect to OpenAI: {e}")
279
+ pass
280
 
281
+ if 'full_text_audio' not in global_dict:
282
+ global_dict.setdefault('full_text_audio', full_text_audio)
283
+ else:
284
+ global_dict['full_text_audio'] = full_text_audio
285
 
286
+ return full_text_audio
287
 
288
 
289
  def get_full_result():