Update app.py

app.py

@@ -142,84 +142,84 @@ def transcribe(audio_filepath, src_lang, tgt_lang, pnc):
 	return output_text
 
 # add logic to make sure dropdown menus only suggest valid combos
-# def on_src_or_tgt_lang_change(src_lang_value, tgt_lang_value, pnc_value):
-# 	"""Callback function for when src_lang or tgt_lang dropdown menus are changed.
-
-# 	Args:
-# 		src_lang_value(string), tgt_lang_value (string), pnc_value(bool) - the current 
-# 			chosen "values" of each Gradio component
-# 	Returns:
-# 		src_lang, tgt_lang, pnc - these are the new Gradio components that will be displayed
-
-# 	"""
-
-# 	if src_lang_value == "English" and tgt_lang_value == "English":
-# 		# src_lang and tgt_lang can go anywhere
-# 		src_lang = gr.Dropdown(
-# 			choices=["English", "Spanish", "French", "German"],
-# 			value=src_lang_value,
-# 			label="Input audio is spoken in:"
-# 		)
-# 		tgt_lang = gr.Dropdown(
-# 			choices=["English", "Spanish", "French", "German"],
-# 			value=tgt_lang_value,
-# 			label="Transcribe in language:"
-# 		)
-# 	elif src_lang_value == "English": 
-# 		# src is English & tgt is non-English
-# 		# => src can only be English or current tgt_lang_values
-# 		# & tgt can be anything
-# 		src_lang = gr.Dropdown(
-# 			choices=["English", tgt_lang_value],
-# 			value=src_lang_value,
-# 			label="Input audio is spoken in:"
-# 		)
-# 		tgt_lang = gr.Dropdown(
-# 			choices=["English", "Spanish", "French", "German"],
-# 			value=tgt_lang_value,
-# 			label="Transcribe in language:"
-# 		)
-# 	elif tgt_lang_value == "English": 
-# 		# src is non-English & tgt is English
-# 		# => src can be anything
-# 		# & tgt can only be English or current src_lang_value
-# 		src_lang = gr.Dropdown(
-# 			choices=["English", "Spanish", "French", "German"],
-# 			value=src_lang_value,
-# 			label="Input audio is spoken in:"
-# 		)
-# 		tgt_lang = gr.Dropdown(
-# 			choices=["English", src_lang_value],
-# 			value=tgt_lang_value,
-# 			label="Transcribe in language:"
-# 		)
-# 	else:
-# 		# both src and tgt are non-English
-# 		# => both src and tgt can only be switch to English or themselves
-# 		src_lang = gr.Dropdown(
-# 			choices=["English", src_lang_value],
-# 			value=src_lang_value,
-# 			label="Input audio is spoken in:"
-# 		)
-# 		tgt_lang = gr.Dropdown(
-# 			choices=["English", tgt_lang_value],
-# 			value=tgt_lang_value,
-# 			label="Transcribe in language:"
-# 		)
-# 	# let pnc be anything if src_lang_value == tgt_lang_value, else fix to True
-# 	if src_lang_value == tgt_lang_value:
-# 		pnc = gr.Checkbox(
-# 			value=pnc_value,
-# 			label="Punctuation & Capitalization in transcript?",
-# 			interactive=True
-# 		)
-# 	else:
-# 		pnc = gr.Checkbox(
-# 			value=True,
-# 			label="Punctuation & Capitalization in transcript?",
-# 			interactive=False
-# 		)
-# 	return src_lang, tgt_lang, pnc
+def on_src_or_tgt_lang_change(src_lang_value, tgt_lang_value, pnc_value):
+	"""Callback function for when src_lang or tgt_lang dropdown menus are changed.
+
+	Args:
+		src_lang_value(string), tgt_lang_value (string), pnc_value(bool) - the current 
+			chosen "values" of each Gradio component
+	Returns:
+		src_lang, tgt_lang, pnc - these are the new Gradio components that will be displayed
+
+	"""
+
+	if src_lang_value == "English" and tgt_lang_value == "English":
+		# src_lang and tgt_lang can go anywhere
+		src_lang = gr.Dropdown(
+			choices=["English", "Spanish", "French", "German"],
+			value=src_lang_value,
+			label="Input audio is spoken in:"
+		)
+		tgt_lang = gr.Dropdown(
+			choices=["English", "Spanish", "French", "German"],
+			value=tgt_lang_value,
+			label="Transcribe in language:"
+		)
+	elif src_lang_value == "English": 
+		# src is English & tgt is non-English
+		# => src can only be English or current tgt_lang_values
+		# & tgt can be anything
+		src_lang = gr.Dropdown(
+			choices=["English", tgt_lang_value],
+			value=src_lang_value,
+			label="Input audio is spoken in:"
+		)
+		tgt_lang = gr.Dropdown(
+			choices=["English", "Spanish", "French", "German"],
+			value=tgt_lang_value,
+			label="Transcribe in language:"
+		)
+	elif tgt_lang_value == "English": 
+		# src is non-English & tgt is English
+		# => src can be anything
+		# & tgt can only be English or current src_lang_value
+		src_lang = gr.Dropdown(
+			choices=["English", "Spanish", "French", "German"],
+			value=src_lang_value,
+			label="Input audio is spoken in:"
+		)
+		tgt_lang = gr.Dropdown(
+			choices=["English", src_lang_value],
+			value=tgt_lang_value,
+			label="Transcribe in language:"
+		)
+	else:
+		# both src and tgt are non-English
+		# => both src and tgt can only be switch to English or themselves
+		src_lang = gr.Dropdown(
+			choices=["English", src_lang_value],
+			value=src_lang_value,
+			label="Input audio is spoken in:"
+		)
+		tgt_lang = gr.Dropdown(
+			choices=["English", tgt_lang_value],
+			value=tgt_lang_value,
+			label="Transcribe in language:"
+		)
+	# let pnc be anything if src_lang_value == tgt_lang_value, else fix to True
+	if src_lang_value == tgt_lang_value:
+		pnc = gr.Checkbox(
+			value=pnc_value,
+			label="Punctuation & Capitalization in transcript?",
+			interactive=True
+		)
+	else:
+		pnc = gr.Checkbox(
+			value=True,
+			label="Punctuation & Capitalization in transcript?",
+			interactive=False
+		)
+	return src_lang, tgt_lang, pnc
 
 
 with gr.Blocks(
@@ -246,22 +246,22 @@ with gr.Blocks(
 
 			gr.HTML("<p><b>Step 2:</b> Choose the input and output language.</p>")
 
-			# src_lang = gr.Dropdown(
-			# 	choices=["English", "Spanish", "French", "German"],
-			# 	value="English",
-			# 	label="Input audio is spoken in:"
-			# )
-
-			# with gr.Column():
-			# 	tgt_lang = gr.Dropdown(
-			# 		choices=["English", "Spanish", "French", "German"],
-			# 		value="English",
-			# 		label="Transcribe in language:"
-			# 	)
-			# 	pnc = gr.Checkbox(
-			# 		value=True,
-			# 		label="Punctuation & Capitalization in transcript?",
-			# 	)
+			src_lang = gr.Dropdown(
+				choices=["English", "Spanish", "French", "German"],
+				value="English",
+				label="Input audio is spoken in:"
+			)
+
+			with gr.Column():
+				tgt_lang = gr.Dropdown(
+					choices=["English", "Spanish", "French", "German"],
+					value="English",
+					label="Transcribe in language:"
+				)
+				pnc = gr.Checkbox(
+					value=True,
+					label="Punctuation & Capitalization in transcript?",
+				)
 
 		with gr.Column():
 
@@ -279,21 +279,21 @@ with gr.Blocks(
 
 	go_button.click(
 		fn=transcribe, 
-		inputs = [audio_file, "English", "English", True],
+		inputs = [audio_file, src_lang, tgt_lang, pnc],
 		outputs = [model_output_text_box]
 	)
 
-	# call on_src_or_tgt_lang_change whenever src_lang or tgt_lang dropdown menus are changed	
-	# src_lang.change(
-	# 	fn=on_src_or_tgt_lang_change,
-	# 	inputs=[src_lang, tgt_lang, pnc],
-	# 	outputs=[src_lang, tgt_lang, pnc],
-	# )
-	# tgt_lang.change(
-	# 	fn=on_src_or_tgt_lang_change,
-	# 	inputs=[src_lang, tgt_lang, pnc],
-	# 	outputs=[src_lang, tgt_lang, pnc],
-	# )
+	call on_src_or_tgt_lang_change whenever src_lang or tgt_lang dropdown menus are changed	
+	src_lang.change(
+		fn=on_src_or_tgt_lang_change,
+		inputs=[src_lang, tgt_lang, pnc],
+		outputs=[src_lang, tgt_lang, pnc],
+	)
+	tgt_lang.change(
+		fn=on_src_or_tgt_lang_change,
+		inputs=[src_lang, tgt_lang, pnc],
+		outputs=[src_lang, tgt_lang, pnc],
+	)
 
 
 demo.queue()