Update app.py

app.py

@@ -80,6 +80,20 @@ def process_input(input_type, user_input, uploaded_file):
         return user_input
     else:
         return "无效的输入类型或未上传文件。"
+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 toggle_visibility(input_type):
     user_input_visible = input_type == "Text Input"