Update app.py

app.py

@@ -1,9 +1,8 @@
 import os  
 import google.generativeai as genai  
-
 from PIL import Image  
 import io  
-from IPython.display import HTML, display  
+import streamlit as st  
 import re  
 
 # Google Gemini API Key  
@@ -59,38 +58,36 @@ def search_and_highlight(full_text, keyword):
     return results, html_text  
 
 def app():  
-    uploaded = files.upload()  
+    st.title("Image OCR and Search")  
+    uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])  
 
-    for filename, file_content in uploaded.items():  
+    if uploaded_file is not None:  
         # Open and display the image  
-        image = Image.open(io.BytesIO(file_content))  
-        display(image)  
+        image = Image.open(uploaded_file)  
+        st.image(image, caption="Uploaded Image", use_column_width=True)  
 
         print("Extracting text from the image...")  
         extracted_text = extract_text_with_gemini(image)  
 
-        print("Extracted Text:")  
-        print(extracted_text)  
+        st.subheader("Extracted Text:")  
+        st.write(extracted_text)  
 
         # Search functionality  
-        while True:  
-            search_keyword = input("\nEnter a keyword to search (or press Enter to exit): ")  
-            if not search_keyword:  
-                break  
-
+        search_keyword = st.text_input("Enter a keyword to search (or press Enter to exit)")  
+        if search_keyword:  
             results, html_text = search_and_highlight(extracted_text, search_keyword)  
 
             if results:  
-                print(f"Keyword '{search_keyword}' found in the extracted text:")  
+                st.subheader(f"Keyword '{search_keyword}' found in the extracted text:")  
                 for i, result in enumerate(results, 1):  
-                    print(f"{i}. ...{result}...")  
+                    st.write(f"{i}. ...{result}...")  
 
                 # Display HTML with highlighted text  
-                display(HTML(f"<p>{html_text}</p>"))  
+                st.markdown(f"<p>{html_text}</p>", unsafe_allow_html=True)  
             else:  
-                print(f"Keyword '{search_keyword}' not found in the extracted text.")  
+                st.write(f"Keyword '{search_keyword}' not found in the extracted text.")  
 
-    print("OCR and search completed.")  
+        st.write("OCR and search completed.")  
 
 if __name__ == "__main__":  
     app()