Update app.py

app.py

@@ -0,0 +1,61 @@
+import streamlit as st
+import pandas as pd
+from utils import extract_text_from_image, speech_to_text
+from ml_model import predict_common_diseases
+from rag_helper import build_knowledge_base, get_medical_guidance
+
+st.set_page_config(page_title="💊 GenAI Drug Finder", layout="centered")
+st.title("💊 GenAI Drug Finder & AI Health Guide")
+
+# Load data
+medicine_df = pd.read_csv("data/medicine_data.csv")
+kb_index, kb_texts, kb_files = build_knowledge_base("data/medical_guides")
+
+# Search medicine
+st.header("🔍 Search for Medicine")
+method = st.radio("Choose Input Method", ["Text Search", "Upload Prescription", "Voice Search"])
+
+query = ""
+
+if method == "Text Search":
+    query = st.text_input("Enter medicine name")
+elif method == "Upload Prescription":
+    file = st.file_uploader("Upload prescription image", type=["png", "jpg", "jpeg"])
+    if file:
+        query = extract_text_from_image(file)
+        st.success(f"Extracted text: {query}")
+elif method == "Voice Search":
+    st.info("Click to record from mic (for local testing only)")
+    if st.button("🎤 Start Recording"):
+        query = speech_to_text()
+        st.success(f"You said: {query}")
+
+if query:
+    results = medicine_df[medicine_df['medicine_name'].str.contains(query, case=False, na=False)]
+    if not results.empty:
+        st.subheader("💊 Matching Medicines")
+        st.dataframe(results)
+    else:
+        st.warning("Medicine not found in inventory.")
+
+# Predict diseases
+st.header("📊 Predict Common Diseases in Area")
+if st.button("📈 Analyze Billing Data"):
+    top_diseases = predict_common_diseases("data/billing_data.csv")
+    st.success("Top Diseases:")
+    for disease in top_diseases:
+        st.markdown(f"✅ {disease}")
+
+# RAG Guidance
+st.header("🧠 AI Health Assistant")
+disease_input = st.selectbox("Select or type a disease", ["diabetes", "cold", "fever", "other"])
+custom_input = ""
+if disease_input == "other":
+    custom_input = st.text_input("Enter disease name")
+
+selected_disease = custom_input if disease_input == "other" else disease_input
+
+if selected_disease:
+    guidance = get_medical_guidance(selected_disease, kb_index, kb_texts)
+    st.subheader("📘 Medical Guidance")
+    st.write(guidance)