Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import os | |
| pwd = os.getcwd() | |
| jpg_files = [f for f in os.listdir(os.path.join(pwd, 'images')) if f.endswith('.jpg')] | |
| # Sort the files by name | |
| jpg_files = sorted(jpg_files, key=lambda x: int(x.split('.')[0])) | |
| txt_files = [f for f in os.listdir(os.path.join(pwd, 'images')) if f.endswith('.txt')] | |
| # Sort the files by name | |
| txt_files = sorted(txt_files, key=lambda x: int(x.split('.')[0])) | |
| # Session state to keep track of the current file index | |
| if 'file_index' not in st.session_state: | |
| st.session_state.file_index = 0 | |
| # Display "Previous" and "Next" buttons | |
| col1, col2 = st.columns(2) | |
| if col1.button("Previous"): | |
| st.session_state.file_index -= 1 | |
| st.session_state.file_index = max(0, st.session_state.file_index) | |
| if col2.button("Next"): | |
| st.session_state.file_index += 1 | |
| st.session_state.file_index = min(len(jpg_files) - 1, st.session_state.file_index) | |
| st.markdown(f"**File {st.session_state.file_index + 1} of {len(jpg_files)}**") | |
| st.text("") | |
| st.text("") | |
| # Display the image and text | |
| jpg_file = jpg_files[st.session_state.file_index] | |
| txt_file = txt_files[st.session_state.file_index] | |
| st.image(os.path.join(pwd, 'images', jpg_file)) | |
| with open(os.path.join(pwd, 'images', txt_file)) as f: | |
| text = f.read() | |
| generated_caption = text.split('\n')[0] | |
| laion_caption = text.split('\n')[1] | |
| st.markdown(f"**Generated Caption:** {generated_caption}") | |
| st.markdown(f"**Laion Caption:** {laion_caption}") |