Update app.py

app.py

@@ -42,6 +42,21 @@ print("Loading the transformer model...")
 # Initialize the transformer vectorizer
 transformer_vectorizer = TransformerVectorizer()
 vectorizer = TfidfVectorizer()
+def process_input(input_type, user_input, uploaded_file):
+    print('ooooocr')
+    if input_type == "File Upload" and uploaded_file is not None:
+        # 读取上传的文件
+        with open(uploaded_file.name, "rb") as f:
+            image = f.read()
+        results = reader.readtext(image)
+        # 提取识别的文本
+        extracted_text = ' '.join([text[1] for text in results])
+        print("提取的文本:")
+        print(extracted_text)
+        return extracted_text
+    elif input_type == "Text Input":
+        return user_input
+'''        
 def process_input(input_type, user_input, uploaded_file):
     if input_type == "File Upload" and uploaded_file is not None:
         file_ext = os.path.splitext(uploaded_file.name)[1].lower()
@@ -80,7 +95,7 @@ def process_input(input_type, user_input, uploaded_file):
         return user_input
     else:
         return "无效的输入类型或未上传文件。"
-
+'''
         
 def toggle_visibility(input_type):
     user_input_visible = input_type == "Text Input"