pages/Captionize.py

import torch 
import re
import langchain
from PIL import Image
import requests
import streamlit as st
from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel 

from PIL import Image
import requests
from langchain.indexes import VectorstoreIndexCreator
from langchain.document_loaders import ImageCaptionLoader


st.set_page_config(page_title="Captionize")

st.title("🤖 Captionize")
st.subheader("Generate Captions for your Image...")

st.sidebar.image('./csv_analysis.png',width=300, use_column_width=True)

# Applying Styling
st.markdown("""
<style>
div.stButton > button:first-child {
    background-color: #0099ff;
    color:#ffffff;
}
div.stButton > button:hover {
    background-color: #00ff00;
    color:#FFFFFF;
    }
</style>""", unsafe_allow_html=True)

image = st.file_uploader("Upload an image", type="jpg")
caption = langchain.generate_caption(image)

button = st.button("Generate Caption")

if button:
    st.write("Caption:", caption)