index.py

import streamlit as st
from PIL import Image
from Model import ImageModel

# Set up the Tesseract path if needed (only for local Tesseract setup)
# pytesseract.pytesseract.tesseract_cmd = r'C:\Path\to\tesseract.exe'
model = ImageModel()
st.title("Image to Text Bot ")
st.subheader("Upload an image and get the Generated text!")

# File uploader
uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])

if uploaded_file is not None:
    # Display the uploaded image
    image = Image.open(uploaded_file)
    st.image(image, caption="Uploaded Image", use_column_width=True)

    extracted_text = model.predict(image)
    st.write("Processing the image...")

    # Extract text using pytesseract
    try:
       
        st.success("Here is the extracted text:")
        st.write(extracted_text)

    except Exception as e:
        st.error(f"An error occurred: {e}")

st.markdown("---")