Update app.py

app.py

@@ -19,7 +19,7 @@ def search_and_highlight(text, keyword):
     highlighted_text = re.sub(f"({keyword})", r"<mark>\1</mark>", text, flags=re.IGNORECASE)
     return highlighted_text
 
-def ocr_and_search(image, keyword):
+def ocr_and_search(image, keyword, languages):
     if image is None:
         return "Please upload an image."
 
@@ -35,12 +35,12 @@ iface = gr.Interface(
     fn=ocr_and_search,
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
-        gr.Textbox(label="Enter keyword to search (optional)")
+        gr.Textbox(label="Enter keyword to search (optional)"),
+        gr.CheckboxGroup(choices=language_choices, label="Select languages", default=["eng"])
     ],
     outputs=gr.HTML(label="Extracted and Highlighted Text"),
     title="OCR and Keyword Search",
     description="Upload an image to extract text using OCR and optionally search for keywords in the extracted text."
 )
 
-if __name__ == '__main__':
-    iface.launch()
+iface.launch()