Spaces:
Sleeping
Sleeping
fix: reduce yield to adapt to mcp
Browse files
app.py
CHANGED
|
@@ -41,21 +41,19 @@ def ocr_workflow_wrapper(file, mistral_key):
|
|
| 41 |
"""
|
| 42 |
if not mistral_key:
|
| 43 |
error_msg = "Error: Mistral API Key not set."
|
| 44 |
-
yield error_msg, error_msg
|
| 45 |
return
|
| 46 |
if not file:
|
| 47 |
error_msg = "Error: File/Text not found."
|
| 48 |
-
yield error_msg, error_msg
|
| 49 |
return
|
| 50 |
|
| 51 |
-
yield "Processing...", "⏳ Processing, please wait...\n\n"
|
| 52 |
-
|
| 53 |
try:
|
| 54 |
result = perform_raw_ocr(file, mistral_key)
|
| 55 |
yield result, f"\n{result}\n"
|
| 56 |
except Exception as e:
|
| 57 |
error_msg = f"An error occurred during processing: {str(e)}"
|
| 58 |
-
yield error_msg, error_msg
|
| 59 |
|
| 60 |
|
| 61 |
def ai_correct(current_text: str, mistral_key: str):
|
|
@@ -71,20 +69,19 @@ def ai_correct(current_text: str, mistral_key: str):
|
|
| 71 |
"""
|
| 72 |
if not mistral_key:
|
| 73 |
error_msg = "Error: Mistral API Key not set."
|
| 74 |
-
yield error_msg, error_msg
|
| 75 |
return
|
| 76 |
if not current_text or current_text.strip() == "":
|
| 77 |
error_msg = "*No text to correct. Upload a file, or paste text into 'Raw Text' box first*"
|
| 78 |
yield error_msg, error_msg
|
| 79 |
return
|
| 80 |
|
| 81 |
-
yield "⏳ AI Correcting text...", "⏳ AI Correcting text...\n\n"
|
| 82 |
try:
|
| 83 |
result = correct_text_with_ai(current_text, mistral_key)
|
| 84 |
yield result, result
|
| 85 |
except Exception as e:
|
| 86 |
error_msg = f"Error : {e}"
|
| 87 |
-
yield error_msg, error_msg
|
| 88 |
|
| 89 |
|
| 90 |
def interpretation_workflow(text: str, genre: str, learn_language: str, target_language: str, gemini_key: str):
|
|
@@ -111,7 +108,6 @@ def interpretation_workflow(text: str, genre: str, learn_language: str, target_l
|
|
| 111 |
yield "Error: Language not selected"
|
| 112 |
|
| 113 |
if genre.lower() in ["general", "news"]:
|
| 114 |
-
yield f"⏳ Generating interpretation for genre: {[genre]} ... (10s - 2min)"
|
| 115 |
result = get_interpretation(genre.lower(), gemini_key, text, learn_language, target_language)
|
| 116 |
yield result
|
| 117 |
else:
|
|
@@ -140,7 +136,6 @@ def translation_workflow(text: str, target_language: str, gemini_key):
|
|
| 140 |
yield "Error: Language not selected"
|
| 141 |
|
| 142 |
if target_language in ["Deutsch", "English", "Français", "Русский язык", "中文"]:
|
| 143 |
-
yield f"⏳ Generating interpretation for target_language: {[target_language]} ..."
|
| 144 |
result = get_translaton(text, gemini_key, target_language)
|
| 145 |
yield result
|
| 146 |
else:
|
|
|
|
| 41 |
"""
|
| 42 |
if not mistral_key:
|
| 43 |
error_msg = "Error: Mistral API Key not set."
|
| 44 |
+
yield error_msg, error_msg
|
| 45 |
return
|
| 46 |
if not file:
|
| 47 |
error_msg = "Error: File/Text not found."
|
| 48 |
+
yield error_msg, error_msg
|
| 49 |
return
|
| 50 |
|
|
|
|
|
|
|
| 51 |
try:
|
| 52 |
result = perform_raw_ocr(file, mistral_key)
|
| 53 |
yield result, f"\n{result}\n"
|
| 54 |
except Exception as e:
|
| 55 |
error_msg = f"An error occurred during processing: {str(e)}"
|
| 56 |
+
yield error_msg, error_msg
|
| 57 |
|
| 58 |
|
| 59 |
def ai_correct(current_text: str, mistral_key: str):
|
|
|
|
| 69 |
"""
|
| 70 |
if not mistral_key:
|
| 71 |
error_msg = "Error: Mistral API Key not set."
|
| 72 |
+
yield error_msg, error_msg
|
| 73 |
return
|
| 74 |
if not current_text or current_text.strip() == "":
|
| 75 |
error_msg = "*No text to correct. Upload a file, or paste text into 'Raw Text' box first*"
|
| 76 |
yield error_msg, error_msg
|
| 77 |
return
|
| 78 |
|
|
|
|
| 79 |
try:
|
| 80 |
result = correct_text_with_ai(current_text, mistral_key)
|
| 81 |
yield result, result
|
| 82 |
except Exception as e:
|
| 83 |
error_msg = f"Error : {e}"
|
| 84 |
+
yield error_msg, error_msg
|
| 85 |
|
| 86 |
|
| 87 |
def interpretation_workflow(text: str, genre: str, learn_language: str, target_language: str, gemini_key: str):
|
|
|
|
| 108 |
yield "Error: Language not selected"
|
| 109 |
|
| 110 |
if genre.lower() in ["general", "news"]:
|
|
|
|
| 111 |
result = get_interpretation(genre.lower(), gemini_key, text, learn_language, target_language)
|
| 112 |
yield result
|
| 113 |
else:
|
|
|
|
| 136 |
yield "Error: Language not selected"
|
| 137 |
|
| 138 |
if target_language in ["Deutsch", "English", "Français", "Русский язык", "中文"]:
|
|
|
|
| 139 |
result = get_translaton(text, gemini_key, target_language)
|
| 140 |
yield result
|
| 141 |
else:
|