Update app.py

app.py

@@ -91,27 +91,16 @@ with st.sidebar:
  st.success("Done")
 
  # Model selection buttons
- st.subheader("Select Model")
+ st.subheader("Select Model:")
+ model_selected = st.session_state.get('model_selected', '')
  col1, col2 = st.columns(2)
- if 'selected_model' not in st.session_state:
- st.session_state.selected_model = None
-
- with col1:
- if st.button("Gemma", key="Gemma", 
- on_click=lambda: setattr(st.session_state, 'selected_model', 'Gemma')):
- st.session_state.selected_model = 'Gemma'
- with col2:
- if st.button("Gemini", key="Gemini", 
- on_click=lambda: setattr(st.session_state, 'selected_model', 'Gemini')):
- st.session_state.selected_model = 'Gemini'
-
- # Apply the selected styling
- if st.session_state.selected_model == 'Gemma':
- col1.button("Gemma", key="reGemma", on_click=None, disabled=True, style="background-color: red;")
- col2.button("Gemini", key="reGemini", on_click=None, disabled=True)
- elif st.session_state.selected_model == 'Gemini':
- col1.button("Gemma", key="reGemma", on_click=None, disabled=True)
- col2.button("Gemini", key="reGemini", on_click=None, disabled=True, style="background-color: red;")
+ if col1.button("GEMMA", key="gemmabtn"):
+ st.session_state.model_selected = "GEMMA"
+ if col2.button("GEMINI", key="geminibtn"):
+ st.session_state.model_selected = "GEMINI"
+ 
+ if model_selected:
+ st.write(f"Selected Model: **{model_selected}**")
 
 user_prompt = st.chat_input("Pregunta acerca del contenido en el archivo PDF:")
 if user_prompt: