Update app.py

app.py

@@ -12,6 +12,28 @@ qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distil
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
+# Main function to interact with the user
+def main():
+    print("Welcome! I'm the UTS Course Chatbot. How can I assist you today?")
+    print("You can ask questions about UTS courses or type 'exit' to end the conversation.")
+
+    while True:
+        user_input = input("You: ")
+        if user_input.lower() == 'exit':
+            print("Bot: Exiting the program.")
+            break
+
+        if "courses" in user_input.lower() and "available" in user_input.lower():
+            field = user_input.split("in ")[-1]
+            courses = data['courses'].get(field, [])
+            if courses:
+                response = f"Courses in {field}: {', '.join(courses)}"
+            else:
+                response = f"No courses found in {field}."
+        else:
+            response = generate_dialog_response(user_input)
+        print("Bot:", response)
+
 # Function to generate response using dialog generation model
 def generate_dialog_response(user_input):
     input_text = user_input + tokenizer.eos_token
@@ -23,39 +45,5 @@ def generate_dialog_response(user_input):
 
     return response_text
 
-# Function to interactively retrieve user input
-def get_user_input():
-    while True:
-        try:
-            # Print prompt and retrieve user input
-            return input("You: ")
-        except EOFError:
-            # If EOFError occurs (e.g., Ctrl+D for EOF on Unix-like systems), inform the user and prompt again
-            print("\nBot: EOFError occurred. Please provide input.")
-
-# Main function to interact with the user
-def main():
-    print("Welcome! I'm the UTS Course Chatbot. How can I assist you today?")
-    print("You can ask questions about UTS courses or type 'exit' to end the conversation.")
-
-    try:
-        while True:
-            user_input = get_user_input()
-            if user_input.lower() == 'exit':
-                print("Bot: Exiting the program.")
-                break
-            elif "courses" in user_input.lower() and "available" in user_input.lower():
-                field = user_input.split("in ")[1]
-                courses = data['courses'].get(field, [])
-                if courses:
-                    response = f"Courses in {field}: {', '.join(courses)}"
-                else:
-                    response = f"No courses found in {field}."
-            else:
-                response = generate_dialog_response(user_input)
-            print("Bot:", response)
-    except KeyboardInterrupt:
-        print("\nBot: Exiting the program.")
-
 if __name__ == "__main__":
     main()