Spaces:
Running
Running
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:
|