Update app.py

app.py

@@ -239,7 +239,7 @@ def run_client():
         # Radio button selection
     analysis_type = st.radio(
         "Choose analysis type:",
-        options=["intervention", "outcome", "question", "db intervention", "db outcome", "db question"],
+        options=["intervention", "outcome", "question", "db intervention", "db outcome"],
         index=2  # Default to 'question'
     )
     
@@ -280,16 +280,10 @@ def run_client():
             relik_model = st.session_state["relik_model"][1]
         elif  analysis_type == "outcome":
             relik_model = st.session_state["relik_model"][2]
-        elif  analysis_type == "db question":
-            relik_model = st.session_state["relik_model"][3]
 
         elif  analysis_type == "db intervention":
             relik_model = st.session_state["relik_model"][4]
 
-        elif  analysis_type == "db outcome":
-            print("hola")
-            relik_model = st.session_state["relik_model"][5]
-
         else:
             relik_model = st.session_state["relik_model"][0]
             
@@ -321,17 +315,23 @@ def run_client():
     
                     # heurstic, prevents split of annotation decorations
                     display = display.replace("border-radius: 0.35em;", "border-radius: 0.35em; white-space: nowrap;")
-
-                with st.container():
-                    st.write(display, unsafe_allow_html=True)
-                
-                text = """
-                <h2 style='color: black;'>Possible Candidates:</h2>
-                <ul style='color: black;'>
-                """ + "".join(f"<li style='color: black;'>{candidate}</li>" for candidate in dict_of_ents_candidates["ents"][2:12]) + "</ul>"
-
-                st.markdown(text, unsafe_allow_html=True)
-
+    
+                    with st.container():
+                        st.write(display, unsafe_allow_html=True)
+                    
+                    text = """
+                    <h2 style='color: black;'>Possible Candidates:</h2>
+                    <ul style='color: black;'>
+                    """ + "".join(f"<li style='color: black;'>{candidate}</li>" for candidate in dict_of_ents_candidates["ents"][2:12]) + "</ul>"
+    
+                    st.markdown(text, unsafe_allow_html=True)
+                else:
+                    text = """
+                    <h2 style='color: black;'>Possible Candidates:</h2>
+                    <ul style='color: black;'>
+                    """ + "".join(f"<li style='color: black;'>{candidate}</li>" for candidate in dict_of_ents_candidates["ents"][0:10]) + "</ul>"
+    
+                    st.markdown(text, unsafe_allow_html=True)
         else:
             st.error("Please enter some text.")