Jorge Henao commited on
Commit
82bd383
1 Parent(s): 0eea790

tabs removed

Browse files
Files changed (2) hide show
  1. app_pinecode.py +14 -35
  2. requirements.txt +0 -1
app_pinecode.py CHANGED
@@ -95,11 +95,11 @@ def search(question, retriever_top_k, reader_top_k):
95
 
96
  return result
97
 
98
- def search_and_show_results(st):
99
  # set start time
100
  stt = time.time()
101
  # retrieve top 5 documents
102
- results = search(query, retriever_top_k=5, reader_top_k=3)
103
  # set endtime
104
  ent = time.time()
105
  # measure resulting time
@@ -123,10 +123,12 @@ def search_and_show_results(st):
123
 
124
  st.markdown("---")
125
 
126
- results = search("que es el adres", retriever_top_k=5, reader_top_k=3)
127
 
128
  # if __name__ == "__main__":
129
  # streamlit part starts here with title
 
 
130
  title = """
131
  <h1 style='
132
  text-align: center;
@@ -145,42 +147,19 @@ st.markdown("""<div align="right">
145
  st.write(f"Cuanto más contexto le des a la pregunta mejor funciona el sistema.")
146
 
147
  # Sidebar
148
- index = None
149
- doc = None
150
  selected_index = None
 
 
 
 
 
 
 
 
151
 
152
  with st.sidebar:
153
  st.write("Ask2Democracy 🇨🇴")
154
  selected_index = st.selectbox("Seleccion la colección sobre el que deseas buscar",
155
  ["Propuesta Reforma de la salud 2023",
156
  "Hallazgos de la comisión de la verdad 2022"])
157
-
158
- tab1, tab2 = st.tabs(["Buscar", "About"])
159
- with tab1:
160
- query = st.text_area("Escribe tu pregunta", on_change=clear_submit)
161
- button = st.button("Buscar")
162
- if button or st.session_state.get("submit"):
163
- if not query:
164
- st.error("¡escribe una pregunta!")
165
- else:
166
- st.session_state["submit"] = True
167
- search_and_show_results(st)
168
-
169
- with tab2:
170
- st.markdown("### Beta disclaimer")
171
- desc = """Sistema experimental en beta. Las respuestas que arroja el sistema no han sido pregrabadas ni basadas en opiniones.
172
- Los documentos indexados han sido tomados de fuentes oficiales. Como el informe final de la CEV y la página oficial de la presidencia de la república de Colombia.
173
- Este explorador usa modelos de inteligencia artificial que ayudan a entender el lenguaje español, sin embargo,
174
- necesita de un mayor entrenamiento por lo que, en ocasiones, puede ser confuso y no tan preciso.
175
- Si quieres apoyar escríbeme a jorge.henao@diezonce.co
176
- """
177
- st.write(desc)
178
- st.write("### Here's how it works:")
179
- st.write("""Usa sentence transformers nombremodelo
180
- nombre modelo reader
181
- pinecode dtabase
182
- Por: Jorge Henao 🇨🇴 Twitter LinkedIn
183
- repo en github con FastAPI / HayStack""")
184
- st.markdown("""---""")
185
- st.write('Author: <a href="https://www.linkedin.com/in/daniel-avila-arias/">Daniel Avila</a>', unsafe_allow_html=True)
186
- st.write('Repo: <a href="https://github.com/davila7/semantic-search">Github</a>', unsafe_allow_html=True)
 
95
 
96
  return result
97
 
98
+ def search_and_show_results():
99
  # set start time
100
  stt = time.time()
101
  # retrieve top 5 documents
102
+ results = fake_search(query, retriever_top_k=5, reader_top_k=3)
103
  # set endtime
104
  ent = time.time()
105
  # measure resulting time
 
123
 
124
  st.markdown("---")
125
 
126
+ #results = search("que es el adres", retriever_top_k=5, reader_top_k=3)
127
 
128
  # if __name__ == "__main__":
129
  # streamlit part starts here with title
130
+
131
+ st.set_page_config(layout="wide", page_title="Buscar")
132
  title = """
133
  <h1 style='
134
  text-align: center;
 
147
  st.write(f"Cuanto más contexto le des a la pregunta mejor funciona el sistema.")
148
 
149
  # Sidebar
 
 
150
  selected_index = None
151
+ query = st.text_area("Escribe tu pregunta", on_change=clear_submit)
152
+ button = st.button("Buscar")
153
+ if button or st.session_state.get("submit"):
154
+ if not query:
155
+ st.error("¡escribe una pregunta!")
156
+ else:
157
+ st.session_state["submit"] = True
158
+ search_and_show_results()
159
 
160
  with st.sidebar:
161
  st.write("Ask2Democracy 🇨🇴")
162
  selected_index = st.selectbox("Seleccion la colección sobre el que deseas buscar",
163
  ["Propuesta Reforma de la salud 2023",
164
  "Hallazgos de la comisión de la verdad 2022"])
165
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
  streamlit==1.15.0
2
- streamlit_chat
3
  sentence-transformers
4
  farm-haystack[pinecone]
5
  pinecone-client
 
1
  streamlit==1.15.0
 
2
  sentence-transformers
3
  farm-haystack[pinecone]
4
  pinecone-client