Update app.py

app.py

@@ -69,7 +69,6 @@ def handle_query(query):
  else:
  return "Disculpa no pude encontrar una respuesta."
 
-
 # Streamlit app initialization
 st.title("(PDF) Chat con documentos de Procesos 🗞️")
 st.markdown("Retrieval-Augmented Generation") 
@@ -91,6 +90,29 @@ with st.sidebar:
  data_ingestion() # Process PDF every time new file is uploaded
  st.success("Done")
 
+ # Model selection buttons
+ st.subheader("Select Model")
+ 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;")
+
 user_prompt = st.chat_input("Pregunta acerca del contenido en el archivo PDF:")
 if user_prompt:
  st.session_state.messages.append({'role': 'user', "content": user_prompt})