import os

# Install PatientSeek from GitHub (only needed once)
os.system("pip install git+https://github.com/whyhow-ai/PatientSeek.git")

import streamlit as st
import pandas as pd
from transformers import pipeline

# Load the model
st.title("PatientSeek AI Model")
st.write("This app uses the PatientSeek model for medical predictions.")

@st.cache_resource  # Cache to speed up loading
def load_model():
    return pipeline("text-classification", model="whyhow-ai/PatientSeek")

model = load_model()

# User input
user_input = st.text_area("Enter patient symptoms:")

if user_input:
    result = model(user_input)
    st.write("Prediction:", result)