Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 model = GPT2LMHeadModel.from_pretrained("gpt2")
 
-# Course information
+# Define course information
 course_info = {
     "Engineering": ["Civil Engineering", "Mechanical Engineering", "Electrical Engineering", "Software Engineering", "etc."],
     "Information Technology": ["Computer Science", "Information Systems", "Cybersecurity", "Data Science", "etc."],
@@ -21,29 +21,27 @@ course_info = {
 
 # Function to generate response
 def generate_response(user_input):
-    # Concatenate course information to the user input
-    input_text = user_input + " ".join([course for category in course_info.values() for course in category])
-
-    # Tokenize input text
-    input_ids = tokenizer.encode(input_text, return_tensors="pt")
-
-    # Generate response
+    input_ids = tokenizer.encode(user_input, return_tensors="pt")
     output = model.generate(input_ids, max_length=100, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
-
-    # Decode and return response
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
+# Function to list courses based on category
+def list_courses(category):
+    if category in course_info:
+        return f"{category} courses include: {', '.join(course_info[category])}"
+    else:
+        return "Sorry, I couldn't find any information on that category."
+
 # Main function to interact with the chatbot
 def chat():
-    print("Welcome to the Course Chatbot! Ask me about available courses or say goodbye to exit.")
+    print("Welcome to the Course Chatbot! Feel free to ask me anything.")
     while True:
         user_input = input("You: ")
-        if user_input.lower() in ["exit", "quit", "goodbye"]:
-            print("Goodbye!")
-            break
+        if any(category in user_input for category in course_info.keys()):
+            bot_response = list_courses(user_input)
         else:
             bot_response = generate_response(user_input)
-            print("Bot:", bot_response)
+        print("Bot:", bot_response)
 
 if __name__ == "__main__":
     chat()