IvanMiao commited on
Commit
e3843d0
·
1 Parent(s): 01dc73d

fix: reduce yield to adapt to mcp

Browse files
Files changed (1) hide show
  1. app.py +5 -10
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 + "\n\n"
45
  return
46
  if not file:
47
  error_msg = "Error: File/Text not found."
48
- yield error_msg, error_msg + "\n\n"
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 + "\n\n"
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 + "\n\n"
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 + "\n\n"
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: