Update app.py

app.py

@@ -38,6 +38,20 @@ 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("text:")
+        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"
     file_upload_visible = input_type == "File Upload"
@@ -287,10 +301,16 @@ with demo:
         "Enter a contract or clause you want to analysis)."
     )
     input_type = gr.Radio(choices=["Text Input", "File Upload"], label="Select Input Method")
-    text = gr.Textbox(label="Enter some words:",visible=False,value="The Employee is entitled to two weeks of paid vacation annually, to be scheduled at the mutual convenience of the Employee and Employer.")
+    user_input = gr.Textbox(label="Enter some words:",visible=False,value="The Employee is entitled to two weeks of paid vacation annually, to be scheduled at the mutual convenience of the Employee and Employer.")
+
+    file_upload = gr.File(label="Upload File", file_types=[".jpg", ".png",".txt"], visible=False)  # Initially hidden
+    input_type.change(toggle_visibility, inputs=input_type, outputs=[user_input, file_upload])
+
+    submit_button = gr.Button("Submit")
+    text = gr.Textbox(label="Extracted Text")
+
+    submit_button.click(process_input, inputs=[input_type, user_input, file_upload], outputs=output_text)
 
-    file_upload = gr.File(label="Upload File", file_types=[".txt"], visible=False)  # Initially hidden
-    input_type.change(toggle_visibility, inputs=input_type, outputs=[text, file_upload])
     gr.Markdown(
         """
 <hr/>