Update app.py

app.py

@@ -815,41 +815,79 @@ if st.session_state.search_results is not None:
                 st.write("") 
                 if foodb_index:
                     with st.popover("🔬 Componentes Moleculares Relevantes"):
+                        
+                        REMEDIAL_PREFIXES = ("anti", "preventive", "remedy", "treatment", "inhibitor")
+
                         st.info("Compuestos en los alimentos mencionados que podrían estar relacionados con tus síntomas.")
+                        
                         user_foods_mentioned = st.session_state.entities.get("alimentos", [])
                         user_symptoms_es = st.session_state.entities.get("sintomas", [])
                         
                         if not user_foods_mentioned or not user_symptoms_es:
                             st.warning("Se necesita al menos un alimento y un síntoma para el análisis molecular.")
                         else:
-                            found_relevant_data = False
+                            found_any_data = False
+                            
                             search_terms_en = []
                             for alimento_es in user_foods_mentioned:
                                 for key_es, value_en_list in FOOD_NAME_TO_FOODB_KEY.items():
                                     if key_es in alimento_es.lower():
                                         search_terms_en.extend(value_en_list)
+                            
                             symptom_keywords_en = []
                             for symptom_es in user_symptoms_es:
                                 if symptom_es in MASTER_SYMPTOM_MAP:
                                     symptom_keywords_en.extend(MASTER_SYMPTOM_MAP[symptom_es].get('keywords_en', []))
+                            
                             for foodb_key, compounds_data in foodb_index.items():
                                 if any(term in foodb_key.lower() for term in set(search_terms_en)):
-                                    relevant_compounds = []
+                                    
+                                    causative_compounds = []
+                                    remedial_compounds = []
+
                                     for item in compounds_data:
                                         effects_en = item.get("effects", [])
-                                        if any(keyword_en in effect.lower() for keyword_en in symptom_keywords_en for effect in effects_en):
-                                            relevant_compounds.append(item)
-                                    if relevant_compounds:
-                                        found_relevant_data = True
+                                        is_causative = False
+                                        is_remedial = False
+                                        
+                                        # Clasificar los efectos
+                                        for effect in effects_en:
+                                            if any(keyword in effect.lower() for keyword in symptom_keywords_en):
+                                                if effect.lower().startswith(REMEDIAL_PREFIXES):
+                                                    is_remedial = True
+                                                else:
+                                                    is_causative = True
+                                        
+                                        if is_causative:
+                                            causative_compounds.append(item)
+                                        elif is_remedial:
+                                            remedial_compounds.append(item)
+
+                                    if causative_compounds or remedial_compounds:
+                                        found_any_data = True
                                         with st.container(border=True):
                                             st.subheader(f"Análisis de: {foodb_key.capitalize()}")
-                                            for item in relevant_compounds[:5]:
-                                                st.write(f"**Compuesto:** {item['compound']}")
-                                                relevant_effects = [eff for eff in item['effects'] if any(kw in eff.lower() for kw in symptom_keywords_en)]
-                                                st.write(f"**Efectos Relevantes (en inglés):** {', '.join(relevant_effects)}")
-                                            st.markdown("---")
-                            if not found_relevant_data:
+
+                                            # Mostrar compuestos causantes primero
+                                            if causative_compounds:
+                                                st.markdown("###### 🔬 Posibles Compuestos Causantes:")
+                                                for item in causative_compounds[:3]:
+                                                    st.write(f"**Compuesto:** {item['compound']}")
+                                                    relevant_effects = [eff for eff in item['effects'] if any(kw in eff.lower() and not eff.lower().startswith(REMEDIAL_PREFIXES) for kw in symptom_keywords_en)]
+                                                    st.write(f"**Efectos Relevantes:** {', '.join(relevant_effects)}")
+                                            
+                                            # Mostrar compuestos remediales después
+                                            if remedial_compounds:
+                                                st.markdown("###### 💊 Compuestos con Efectos Potencialmente Beneficiosos:")
+                                                for item in remedial_compounds[:3]:
+                                                    st.write(f"**Compuesto:** {item['compound']}")
+                                                    relevant_effects = [eff for eff in item['effects'] if any(kw in eff.lower() and eff.lower().startswith(REMEDIAL_PREFIXES) for kw in symptom_keywords_en)]
+                                                    st.write(f"**Efectos Relevantes:** {', '.join(relevant_effects)}")
+                                        st.markdown("---")
+
+                            if not found_any_data:
                                 st.warning("No se encontraron compuestos moleculares directamente relacionados con tus síntomas para los alimentos mencionados.")
+            
             st.markdown("---")
 
             with st.container(border=True):