Update a.py

a.py

@@ -98,6 +98,25 @@ def truncate_at_full_stop(text, max_length=1024):
     return truncated
 
 
+df1 = pd.read_csv(
+    'med_name.csv', encoding='utf-8'
+)
+
+# Convert the 'Meds' column to a lowercase list
+KNOWN_MEDICINES = df1['Meds   '].str.lower().tolist()
+
+
+def contains_medicine_terms(output):
+    """Check if the output contains terms that indicate a medicine name."""
+    return any(term in output for term in [" IP ", " mg ", " ml ", " Mg ", " Ml ", "mg ", "ml ",
+                                           " gm ", "gm ", " mcg ", "mcg "])
+
+
+def is_valid_medicine_in_input(user_input):
+    """Check if the user input contains any valid medicine name."""
+    return any(med in user_input.lower() for med in KNOWN_MEDICINES)
+
+
 if st.session_state.selected_service == "Kendr Locator":
     st.title("MedBot")
 
@@ -107,13 +126,27 @@ if st.session_state.selected_service == "Kendr Locator":
         if user_input:
             if contains_profanity(user_input):
                 st.markdown("<span style='color: red;'>Mind The Language Dear!</span>", unsafe_allow_html=True)
+
             else:
                 formatted_prompt = template.format(user_input, "")
 
                 response = llm.invoke(formatted_prompt)
 
-                truncated_response = truncate_at_full_stop(response)
-                st.markdown(f"**MedBot:** {truncated_response}", unsafe_allow_html=False)
-
-
-st.warning("Developer's notice : Responses are generated by AI and maybe inaccurate or inappropriate. Any received medical or financial consult is not a substitute for professional advice.")
+                if contains_medicine_terms(response):
+                    # Generated response has medical terms
+                    if is_valid_medicine_in_input(user_input):
+                        truncated_response = truncate_at_full_stop(response)
+                        st.markdown(f"**MedBot:** {truncated_response}", unsafe_allow_html=False)
+                    else:
+                        st.markdown("<span style='color: green;'>"
+                                    "Please consult to a Pharmacist at you nearest Janaushadi Kendr""<br>"
+                                    "Use Kendr Locator to locate one near you!"
+                                    "</span>", unsafe_allow_html=True)
+
+                else:
+                    # No medicine-related terms, safe to display response
+                    truncated_response = truncate_at_full_stop(response)
+                    st.markdown(f"**MedBot:** {truncated_response}", unsafe_allow_html=False)
+
+st.warning("Developer's notice : Responses are generated by AI and maybe inaccurate or inappropriate."
+           "Any received medical or financial consult is not a substitute for professional advice.")