fix: typo, error handle

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from process.ocr import perform_raw_ocr, correct_text_with_ai
 from process.interpretation import get_interpretation
-from process.translation import get_tranlaton
+from process.translation import get_translaton
 from process.gradio_css import CUSTOM_CSS
 
 
@@ -78,7 +78,7 @@ def ai_correct(current_text: str, mistral_key: str):
 		yield error_msg, error_msg
 		return
 
-	yield "⏳ AI Correcting text...", "⏳ AI Correcting text...\n\n*Please wait...*"
+	yield "⏳ AI Correcting text...", "⏳ AI Correcting text...\n\n"
 	try:
 		result = correct_text_with_ai(current_text, mistral_key)
 		yield result, result
@@ -141,7 +141,7 @@ def translation_workflow(text: str, target_language: str, gemini_key):
 	
 	if target_language in ["Deutsch", "English", "Français", "Русский язык", "中文"]:
 		yield f"⏳ Generating interpretation for target_language: {[target_language]} ..."
-		result = get_tranlaton(text, gemini_key, target_language)
+		result = get_translaton(text, gemini_key, target_language)
 		yield result
 	else:
 		yield "not implemented yet"
@@ -206,6 +206,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=CUSTOM_CSS) as demo:
 					with gr.Tab("Formatted Text"):
 						text_markdown = gr.Markdown(
 							value="*Processed text will appear here...*\n\n",
+							label="Formatted Text"
 						)
 
 	# Hook the ocr button to click event
@@ -237,6 +238,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=CUSTOM_CSS) as demo:
 				gr.Markdown("### COURSE")
 				interpretation_output = gr.Markdown(
 					value="*Interpretation will appear here after processing...*\n\n",
+					label="Interpretation Result",
 					show_copy_button=True
 					)
 
@@ -260,6 +262,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=CUSTOM_CSS) as demo:
 			with gr.Column(scale=2):
 				interpretation_output = gr.Markdown(
 					value="*Translation will appear here ...*\n\n",
+					label="Translation Result",
 					show_copy_button=True
 					)
 

process/interpretation.py

@@ -18,7 +18,10 @@ def get_interpretation(genre: str,
 	if not text:
 		return "Error: text not found."
 
-	client = genai.Client(api_key=api_key)
+	try:
+		client = genai.Client(api_key=api_key)
+	except Exception as e:
+		return f"ERROR: {str(e)}"
 
 	lang_map ={"DE": "German", "EN": "English", "FR": "French", "RU":"Russian", "ZH": "Chinese"}
 	learn_lang = lang_map.get(learn_language.upper(), "English")

process/ocr.py

@@ -6,11 +6,16 @@ OCR_MODEL = "mistral-ocr-latest"
 CHAT_MODEL = "mistral-large-latest"
 
 
-def ocr_from_file(file_path, api_key, mode="image"):
+def ocr_from_file(file_path, api_key: str, mode="image"):
 
 	if not api_key:
 		raise ValueError("Mistral API Key is required.")
-	client = Mistral(api_key=api_key)
+
+	try:
+		client = Mistral(api_key=api_key)
+	except Exception as e:
+		raise ValueError("API invalid.")
+
 	uploaded_image = client.files.upload(
 		file={
 			"file_name": file_path,
@@ -51,11 +56,15 @@ def get_combined_markdown(ocr_response: OCRResponse) -> str:
 	return "\n\n".join(markdowns)
 
 
-def correct_text_with_ai(text: str, api_key: str):
+def correct_text_with_ai(text: str, api_key: str) -> str:
 
 	if not api_key:
 		raise ValueError("Mistral API Key is required.")
-	client = Mistral(api_key=api_key)
+
+	try:
+		client = Mistral(api_key=api_key)
+	except Exception as e:
+		return f"ERROR: {str(e)}"
 
 	response = client.chat.complete(
 		model=CHAT_MODEL,
@@ -85,7 +94,7 @@ def correct_text_with_ai(text: str, api_key: str):
 	return(response.choices[0].message.content)
 
 
-def perform_raw_ocr(input_file, api_key):
+def perform_raw_ocr(input_file, api_key: str):
 	if input_file != None:
 		file_ext = input_file.name.split('.')[-1].lower()
 	else:

process/translation.py

@@ -9,14 +9,17 @@ Do not add any extra information, explanations, or stylistic changes.
 Maintain the original meaning and tone as closely as possible.
 """
 
-def get_tranlaton(text: str, api_key: str, target_language: str) -> str:
+def get_translaton(text: str, api_key: str, target_language: str) -> str:
 
 	if not api_key:
 		return "Error: Gemini API Key not found."
 	if not text:
 		return "Error: text not found."
 
-	client = genai.Client(api_key=api_key)
+	try:
+		client = genai.Client(api_key=api_key)
+	except Exception as e:
+		return f"ERROR: {str(e)}"
 
 	lang_map = {"Deutsch": "German", "English": "English", "Français": "French", "Русский язык": "Russain", "中文": "Chinese"}
 	tar_lang = lang_map.get(target_language, "English")