Update app.py

app.py

@@ -1,22 +1,28 @@
-import streamlit as st
 from datasets import load_dataset
 
 # Load the dataset
 dataset = load_dataset("marksverdhei/wordnet-definitions-en-2021")
 
-# Streamlit app
-st.title("TouchDown Buddy Chatbot")
-
-# Input question
-question = st.text_input("Enter your question:")
-
-# Search for the corresponding definition
-if question:
+def get_definition(word):
     found = False
     for example in dataset["train"]:
-        if question.lower() in example["word"].lower():
-            st.write("Definition:", example["gloss"])
+        if word.lower() in example["word"].lower():
+            return example["gloss"]
             found = True
             break
     if not found:
-        st.write("Sorry, the answer to your question is not available.")
+        return "Sorry, the definition for '{}' is not available.".format(word)
+
+def main():
+    while True:
+        user_input = input("Enter your question (or type 'quit' to exit): ")
+        if user_input.lower() == "quit":
+            print("Goodbye!")
+            break
+        else:
+            word_to_define = user_input.split()[-1]  # Get the last word from the user input
+            definition = get_definition(word_to_define)
+            print("Definition:", definition)
+
+if __name__ == "__main__":
+    main()