pskeshu commited on
Commit
50f6d31
·
1 Parent(s): b329e55

updated streamlite version, random button functionality

Browse files
Files changed (2) hide show
  1. app.py +18 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -84,14 +84,30 @@ if sample_images_available:
84
  use_source = st.sidebar.radio("Select Source:", ["Sample Images", "Upload Image"])
85
 
86
  if use_source == "Sample Images":
 
 
 
 
87
  # Image selector
88
  image_names = [img.name for img in sample_images]
89
- selected_image = st.sidebar.selectbox("Choose Sample Image:", image_names)
90
 
 
91
  if st.sidebar.button("🎲 Random Sample Image"):
92
- selected_image = random.choice(image_names)
93
  st.rerun()
94
 
 
 
 
 
 
 
 
 
 
 
 
 
95
  sample_image_path = sample_images_path / selected_image
96
  uploaded_file = None
97
  else:
 
84
  use_source = st.sidebar.radio("Select Source:", ["Sample Images", "Upload Image"])
85
 
86
  if use_source == "Sample Images":
87
+ # Initialize session state for selected image
88
+ if 'selected_sample_image' not in st.session_state:
89
+ st.session_state.selected_sample_image = sample_images[0].name
90
+
91
  # Image selector
92
  image_names = [img.name for img in sample_images]
 
93
 
94
+ # Random button
95
  if st.sidebar.button("🎲 Random Sample Image"):
96
+ st.session_state.selected_sample_image = random.choice(image_names)
97
  st.rerun()
98
 
99
+ # Selectbox with session state
100
+ selected_image = st.sidebar.selectbox(
101
+ "Choose Sample Image:",
102
+ image_names,
103
+ index=image_names.index(st.session_state.selected_sample_image) if st.session_state.selected_sample_image in image_names else 0,
104
+ key="sample_image_selector"
105
+ )
106
+
107
+ # Update session state if user manually changes selection
108
+ if selected_image != st.session_state.selected_sample_image:
109
+ st.session_state.selected_sample_image = selected_image
110
+
111
  sample_image_path = sample_images_path / selected_image
112
  uploaded_file = None
113
  else:
requirements.txt CHANGED
@@ -8,7 +8,7 @@ requests
8
  google-generativeai
9
  anthropic
10
  python-dotenv
11
- streamlit==1.30.0
12
  plotly
13
  networkx
14
  owlready2
 
8
  google-generativeai
9
  anthropic
10
  python-dotenv
11
+ streamlit
12
  plotly
13
  networkx
14
  owlready2