Younesse Kaddar
update
e867a2d
raw
history blame
2.41 kB
import streamlit as st
# Import the LangChain library
import langchain
# Load the AI model
model = langchain.load_model("model.pkl")
# Create a function to get the feedback from the AI model
def get_feedback(statement):
# Get the predictions from the AI model
predictions = model.predict(statement)
# Create a list of feedback
feedback = []
for prediction in predictions:
feedback.append(prediction["feedback"])
return feedback
# Create a function to display the feedback
def display_feedback(statement):
# Get the feedback from the AI model
feedback = get_feedback(statement)
# Display the feedback to the user
st.write("Here is the feedback from the AI model:")
st.write(feedback)
# Create a main function
def main():
# Get the personal statement from the user
statement = st.text_input("Enter your personal statement:")
# Display the feedback to the user
display_feedback(statement)
# Run the main function
if __name__ == "__main__":
main()
# print("Start!")
# load_dotenv(find_dotenv())
# # pinecone.init(api_key=os.getenv("PINECONE_API_KEY"), environment=os.getenv("PINECONE_ENVIRONMENT"))
# dataset_path = "./dataset.txt"
# loader = TextLoader(dataset_path)
# comments = loader.load_and_split()
# embeddings = OpenAIEmbeddings(model_name="ada")
# vectordb = Chroma.from_documents(comments, embedding=embeddings, persist_directory=".")
# vectordb.persist()
# memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
# # Assuming that GPT-4 is used for grammar, structure, and fact-checking
# # and Claude is used for providing tips and encouraging students to do their own research
# grammar_llm = OpenAI(temperature=0.8)
# tips_llm = Claude(temperature=0.8)
# grammar_qa = ConversationalRetrievalChain.from_llm(grammar_llm, vectordb.as_retriever(), memory=memory)
# tips_qa = ConversationalRetrievalChain.from_llm(tips_llm, vectordb.as_retriever(), memory=memory)
# st.title('AI Statement Reviewer')
# user_input = st.text_area("Enter your personal statement here:")
# if st.button('Get feedback'):
# grammar_result = grammar_qa({"question": user_input})
# tips_result = tips_qa({"question": user_input})
# st.write("Grammar and Structure Feedback:")
# st.write(grammar_result["answer"])
# st.write("Tips and Recommendations:")
# st.write(tips_result["answer"])