import streamlit as st
import ollama as ol
import time

desiredModel = "deepseek-r1:1.5b"

st.title("protoype app using deepseek-r1:1.5b")

def generate_response(questionToAsk):
    response = ol.chat(model=desiredModel, messages=[
                {
                    'role': 'user',
                    'content' : questionToAsk,
                }
            ]
        )
    st.info(response['message']['content'])

with st.form("my_form"):
    text = st.text_area(
        "Enter text:",
    )
    st.write("Would you like to be reminded in 5 minutes?")
    checkbox_val = st.checkbox("Yes")
    submitted = st.form_submit_button("Submit")
    
    if submitted:
        with st.status("Reminder in 5 minutes..."):
            time.sleep(300)
            st.write("All done!")
        generate_response(text)