Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from langchain.prompts import ChatPromptTemplate
 from langchain_huggingface import HuggingFaceEndpoint
-from langchain.schema import HumanMessage, SystemMessage
+from langchain.schema import AIMessage, HumanMessage, SystemMessage
 import os
 import time
 import logging
@@ -54,7 +54,7 @@ Provide detailed, actionable advice with specific examples."""),
 study_template = ChatPromptTemplate.from_messages([
     ("system", """{system_message}
 
-You are a study skills coach. Help students with:
+You are a study skills coach. Help students with these types of tasks:
 - Effective study methods for different learning styles
 - Time management and scheduling techniques
 - Memory techniques and retention strategies
@@ -289,7 +289,7 @@ with gr.Blocks(css=custom_css, title="EduBot") as demo:
             msg = gr.Textbox(
                 placeholder="Ask me about math, research, study strategies, or any educational topic...",
                 show_label=False,
-                lines=2,
+                lines=4,
                 elem_classes=["input-container"],
                 scale=4
             )