app.py

import streamlit as st
st.title('AI Doctor')

AA=st.text_input('Enter Your Symptoms')
submit=st.button('SUBMIT')

from langchain_groq import ChatGroq
model=ChatGroq(temperature=0.6,
               groq_api_key='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',
               model_name="llama-3.1-70b-versatile")

from langchain.prompts import PromptTemplate
query=PromptTemplate(input_varibles=['x'],
                     template='I am having a{x} please suggest me a treatment in 20 words')

if submit:
    B=query.format(x=AA)
    response=model.invoke(B)
    st.write(response.content)