Update app.py

app.py

@@ -7,23 +7,20 @@ import time
 import logging
 import re
 
-# Set up logging for better debugging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Set up LangChain model with conservative settings
 llm = HuggingFaceEndpoint(
-    repo_id="HuggingFaceH4/zephyr-7b-beta",
-    temperature=0.7,
-    top_p=0.9,
-    repetition_penalty=1.1,
-    model_kwargs={"max_length": 1024},
-    huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
+   repo_id="HuggingFaceH4/zephyr-7b-beta",
+   temperature=0.7,
+   top_p=0.9,
+   repetition_penalty=1.1,
+   model_kwargs={"max_length": 1024},
+   huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
 )
 
-# Enhanced prompt templates that use system_message parameter
 math_template = ChatPromptTemplate.from_messages([
-    ("system", """{system_message}
+   ("system", """{system_message}
 
 You are an expert math tutor. For every math problem:
 1. Break it down into key concepts
@@ -31,11 +28,11 @@ You are an expert math tutor. For every math problem:
 3. Outline the process for solving a similar problem
 
 Be comprehensive and educational. Structure your response clearly."""),
-    ("human", "{question}")
+   ("human", "{question}")
 ])
 
 research_template = ChatPromptTemplate.from_messages([
-    ("system", """{system_message}
+   ("system", """{system_message}
 
 You are a research skills mentor. Help students with:
 - Determining the validity of sources
@@ -46,11 +43,11 @@ You are a research skills mentor. Help students with:
 - Database navigation and search strategies
 
 Provide detailed, actionable advice with specific examples."""),
-    ("human", "{question}")
+   ("human", "{question}")
 ])
 
 study_template = ChatPromptTemplate.from_messages([
-    ("system", """{system_message}
+   ("system", """{system_message}
 
 You are a study skills coach. Help students with:
 - Effective study methods for different learning styles
@@ -62,11 +59,11 @@ You are a study skills coach. Help students with:
 - Offer short quiz sessions where you pose one to two questions at a time, then provide feedback on the students answers.
 
 Provide comprehensive, personalized advice with practical examples."""),
-    ("human", "{question}")
+   ("human", "{question}")
 ])
 
 general_template = ChatPromptTemplate.from_messages([
-    ("system", """{system_message}
+   ("system", """{system_message}
 
 You are EduBot, a comprehensive AI learning assistant. You help students with:
 📐 Mathematics (Concise explainations rooted in understanding the concepts and process rather than answering the math problem directly)
@@ -75,295 +72,276 @@ You are EduBot, a comprehensive AI learning assistant. You help students with:
 🛠️ Educational tools (guidance on learning resources and technologies)
 
 Always be encouraging, patient, thorough, and comprehensive."""),
-    ("human", "{question}")
+   ("human", "{question}")
 ])
 
 def detect_subject(message):
-    """Determine which prompt template to use based on the message"""
-    message_lower = message.lower()
-    
-    math_keywords = ['math', 'solve', 'calculate', 'equation', 'formula', 'algebra', 'geometry', 'calculus', 'derivative', 'integral', 'theorem', 'proof']
-    research_keywords = ['research', 'source', 'citation', 'bibliography', 'reference', 'academic', 'paper', 'essay', 'thesis', 'database', 'journal']
-    study_keywords = ['study', 'memorize', 'exam', 'test', 'quiz', 'review', 'learn', 'remember', 'focus', 'motivation', 'notes']
-    
-    if any(keyword in message_lower for keyword in math_keywords):
-        return math_template, "🧮 Math Mode"
-    elif any(keyword in message_lower for keyword in research_keywords):
-        return research_template, "🔍 Research Mode"
-    elif any(keyword in message_lower for keyword in study_keywords):
-        return study_template, "📚 Study Mode"
-    else:
-        return general_template, "🎓 General Mode"
+   message_lower = message.lower()
+   
+   math_keywords = ['math', 'solve', 'calculate', 'equation', 'formula', 'algebra', 'geometry', 'calculus', 'derivative', 'integral', 'theorem', 'proof']
+   research_keywords = ['research', 'source', 'citation', 'bibliography', 'reference', 'academic', 'paper', 'essay', 'thesis', 'database', 'journal']
+   study_keywords = ['study', 'memorize', 'exam', 'test', 'quiz', 'review', 'learn', 'remember', 'focus', 'motivation', 'notes']
+   
+   if any(keyword in message_lower for keyword in math_keywords):
+       return math_template, "🧮 Math Mode"
+   elif any(keyword in message_lower for keyword in research_keywords):
+       return research_template, "🔍 Research Mode"
+   elif any(keyword in message_lower for keyword in study_keywords):
+       return study_template, "📚 Study Mode"
+   else:
+       return general_template, "🎓 General Mode"
 
 def smart_truncate(text, max_length=3000):
-    """Intelligently truncate text at sentence boundaries"""
-    if len(text) <= max_length:
-        return text
-    
-    # Try to cut at last complete sentence
-    sentences = re.split(r'(?<=[.!?]) +', text[:max_length])
-    if len(sentences) > 1:
-        # Remove the last incomplete sentence
-        return ' '.join(sentences[:-1]) + "... [Response truncated - ask for continuation]"
-    else:
-        # Fallback to word boundary
-        words = text[:max_length].split()
-        return ' '.join(words[:-1]) + "... [Response truncated - ask for continuation]"
+   if len(text) <= max_length:
+       return text
+   
+   sentences = re.split(r'(?<=[.!?]) +', text[:max_length])
+   if len(sentences) > 1:
+       return ' '.join(sentences[:-1]) + "... [Response truncated - ask for continuation]"
+   else:
+       words = text[:max_length].split()
+       return ' '.join(words[:-1]) + "... [Response truncated - ask for continuation]"
 
 def respond_with_enhanced_streaming(
-    message,
-    history,
-    max_tokens=600,
-    temperature=0.7,
-    top_p=0.9,
+   message,
+   history,
+   max_tokens=600,
+   temperature=0.7,
+   top_p=0.9,
 ):
-    """Enhanced LangChain implementation with proper system message handling"""
-    
-    try:
-        # Select template and get mode
-        template, mode = detect_subject(message)
-        
-        # Create LangChain chain
-        chain = template | llm
-        
-        # Show initial mode
-        yield f"*{mode}*\n\nGenerating response..."
-        
-        # Get complete response from LangChain with system message
-        logger.info(f"Processing {mode} query: {message[:50]}...")
-        
-        response = chain.invoke({
-            "question": message,
-            "system_message": "You are EduBot, an expert AI learning assistant. Provide comprehensive, educational responses that help students truly understand concepts."
-        })
-        
-        # Smart truncation at sentence boundaries
-        response = smart_truncate(response, max_length=3000)
-        
-        # Simulate streaming by chunking the response
-        words = response.split()
-        partial_response = f"*{mode}*\n\n"
-        
-        # Stream word by word for better UX
-        for i, word in enumerate(words):
-            partial_response += word + " "
-            
-            # Update every 4 words for smooth streaming effect
-            if i % 4 == 0:
-                yield partial_response
-                time.sleep(0.03)
-        
-        # Final complete response
-        final_response = f"*{mode}*\n\n{response}"
-        logger.info(f"Response completed. Length: {len(response)} characters")
-        yield final_response
-        
-    except Exception as e:
-        logger.exception("Error in LangChain response generation")
-        yield f"Sorry, I encountered an error: {str(e)[:150]}"
+   try:
+       template, mode = detect_subject(message)
+       chain = template | llm
+       
+       yield f"*{mode}*\n\nGenerating response..."
+       
+       logger.info(f"Processing {mode} query: {message[:50]}...")
+       
+       response = chain.invoke({
+           "question": message,
+           "system_message": "You are EduBot, an expert AI learning assistant. Provide comprehensive, educational responses that help students truly understand concepts."
+       })
+       
+       response = smart_truncate(response, max_length=3000)
+       
+       words = response.split()
+       partial_response = f"*{mode}*\n\n"
+       
+       for i, word in enumerate(words):
+           partial_response += word + " "
+           
+           if i % 4 == 0:
+               yield partial_response
+               time.sleep(0.03)
+       
+       final_response = f"*{mode}*\n\n{response}"
+       logger.info(f"Response completed. Length: {len(response)} characters")
+       yield final_response
+       
+   except Exception as e:
+       logger.exception("Error in LangChain response generation")
+       yield f"Sorry, I encountered an error: {str(e)[:150]}"
 
-# Your custom Glass theme with amber accent
 theme = gr.themes.Glass(
-    primary_hue="amber",
-    text_size="md",
-    radius_size="md",
+   primary_hue="amber",
+   text_size="md",
+   radius_size="md",
 ).set(
-    shadow_drop='*shadow_drop_lg',
-    shadow_drop_lg='*shadow_drop',
-    shadow_spread='*shadow_drop_lg',
-    block_info_text_weight='500',
-    button_border_width='*block_border_width'
+   shadow_drop='*shadow_drop_lg',
+   shadow_drop_lg='*shadow_drop',
+   shadow_spread='*shadow_drop_lg',
+   block_info_text_weight='500',
+   button_border_width='*block_border_width'
 )
 
-# Responsive interface with flexible height and custom styling
 with gr.Blocks(theme=theme, css="""
-    .main-container {
-        max-width: 900px;
-        margin: 0 auto;
-        padding: 1rem;
-        height: 100vh;
-        display: flex;
-        flex-direction: column;
-    }
-    .title {
-        text-align: center;
-        font-size: clamp(1.8rem, 4vw, 2.5rem);
-        font-weight: bold;
-        color: #d97706;
-        margin-bottom: 1rem;
-        flex-shrink: 0;
-    }
-    .chat-container {
-        flex: 1;
-        display: flex;
-        flex-direction: column;
-        min-height: 0;
-        margin-bottom: 1rem;
-    }
-    .gradio-chatbot {
-        flex: 1 !important;
-        height: auto !important;
-        min-height: 400px;
-        max-height: calc(100vh - 200px);
-    }
-    
-    /* Hide the chat icon */
-    .chat-bot .message-wrap .avatar-container {
-        display: none !important;
-    }
-    .gradio-chatbot .message-wrap .avatar-container {
-        display: none !important;
-    }
-    .gradio-chatbot .message.bot .avatar-container,
-    .gradio-chatbot .message.user .avatar-container {
-        display: none !important;
-    }
-    
-    /* Style the input area */
-    .input-row {
-        flex-shrink: 0;
-        margin-top: 0.5rem;
-        display: flex;
-        justify-content: center;
-        align-items: center;
-        gap: 10px;
-        max-width: 700px;
-        margin-left: auto;
-        margin-right: auto;
-    }
-    
-    /* Custom textbox styling */
-    .custom-textbox input {
-        border-radius: 25px !important;
-        border: 2px solid #64748b !important;
-        padding: 15px 20px !important;
-        font-size: 16px !important;
-        background: white !important;
-        outline: none !important;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
-        transition: all 0.3s ease !important;
-    }
-    
-    .custom-textbox input:focus {
-        border-color: #475569 !important;
-        box-shadow: 0 0 0 3px rgba(100, 116, 139, 0.1) !important;
-    }
-    
-    /* Remove textbox label */
-    .custom-textbox label {
-        display: none !important;
-    }
-    
-    /* Center the send button */
-    .send-button {
-        border-radius: 50% !important;
-        width: 50px !important;
-        height: 50px !important;
-        display: flex !important;
-        align-items: center !important;
-        justify-content: center !important;
-        flex-shrink: 0 !important;
-    }
-    
-    /* Mobile responsiveness */
-    @media (max-width: 768px) {
-        .main-container {
-            padding: 0.5rem;
-        }
-        .title {
-            font-size: 1.5rem;
-            margin-bottom: 0.5rem;
-        }
-        .gradio-chatbot {
-            min-height: 300px;
-            max-height: calc(100vh - 150px);
-        }
-        .input-row {
-            margin-top: 0.25rem;
-            max-width: 100%;
-            padding: 0 0.5rem;
-        }
-        .custom-textbox input {
-            padding: 12px 15px !important;
-            font-size: 14px !important;
-        }
-    }
-    
-    /* Tablet responsiveness */
-    @media (min-width: 769px) and (max-width: 1024px) {
-        .gradio-chatbot {
-            min-height: 450px;
-            max-height: calc(100vh - 180px);
-        }
-    }
-    
-    /* Large screens */
-    @media (min-width: 1025px) {
-        .gradio-chatbot {
-            min-height: 500px;
-            max-height: calc(100vh - 200px);
-        }
-    }
-    
-    /* Landscape mobile */
-    @media (max-height: 500px) and (orientation: landscape) {
-        .title {
-            font-size: 1.2rem;
-            margin-bottom: 0.25rem;
-        }
-        .gradio-chatbot {
-            min-height: 200px;
-            max-height: calc(100vh - 100px);
-        }
-    }
+   .main-container {
+       max-width: 900px;
+       margin: 0 auto;
+       padding: 1rem;
+       height: 100vh;
+       display: flex;
+       flex-direction: column;
+   }
+   .title {
+       text-align: center;
+       font-size: clamp(1.8rem, 4vw, 2.5rem);
+       font-weight: bold;
+       color: #d97706;
+       margin-bottom: 1rem;
+       flex-shrink: 0;
+   }
+   .chat-container {
+       flex: 1;
+       display: flex;
+       flex-direction: column;
+       min-height: 0;
+       margin-bottom: 1rem;
+   }
+   .gradio-chatbot {
+       flex: 1 !important;
+       height: auto !important;
+       min-height: 400px;
+       max-height: calc(100vh - 200px);
+   }
+   
+   .chat-bot .message-wrap .avatar-container {
+       display: none !important;
+   }
+   .gradio-chatbot .message-wrap .avatar-container {
+       display: none !important;
+   }
+   .gradio-chatbot .message.bot .avatar-container,
+   .gradio-chatbot .message.user .avatar-container {
+       display: none !important;
+   }
+   
+   .input-row {
+       flex-shrink: 0;
+       margin-top: 0.5rem;
+       display: flex;
+       justify-content: center;
+       align-items: center;
+       gap: 10px;
+       max-width: 700px;
+       margin-left: auto;
+       margin-right: auto;
+   }
+   
+   .custom-textbox {
+       min-width: 500px !important;
+       width: 100% !important;
+   }
+   
+   .custom-textbox input {
+       border-radius: 25px !important;
+       border: 2px solid #64748b !important;
+       padding: 15px 20px !important;
+       font-size: 16px !important;
+       background: white !important;
+       outline: none !important;
+       box-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
+       transition: all 0.3s ease !important;
+       min-width: 500px !important;
+       min-height: 50px !important;
+       height: 50px !important;
+   }
+   
+   .custom-textbox input:focus {
+       border-color: #475569 !important;
+       box-shadow: 0 0 0 3px rgba(100, 116, 139, 0.1) !important;
+   }
+   
+   .custom-textbox label {
+       display: none !important;
+   }
+   
+   .send-button {
+       border-radius: 50% !important;
+       width: 50px !important;
+       height: 50px !important;
+       display: flex !important;
+       align-items: center !important;
+       justify-content: center !important;
+       flex-shrink: 0 !important;
+   }
+   
+   @media (max-width: 768px) {
+       .main-container {
+           padding: 0.5rem;
+       }
+       .title {
+           font-size: 1.5rem;
+           margin-bottom: 0.5rem;
+       }
+       .gradio-chatbot {
+           min-height: 300px;
+           max-height: calc(100vh - 150px);
+       }
+       .input-row {
+           margin-top: 0.25rem;
+           max-width: 100%;
+           padding: 0 0.5rem;
+       }
+       .custom-textbox {
+           min-width: 300px !important;
+       }
+       .custom-textbox input {
+           padding: 12px 15px !important;
+           font-size: 14px !important;
+           min-width: 300px !important;
+           min-height: 45px !important;
+           height: 45px !important;
+       }
+   }
+   
+   @media (min-width: 769px) and (max-width: 1024px) {
+       .gradio-chatbot {
+           min-height: 450px;
+           max-height: calc(100vh - 180px);
+       }
+   }
+   
+   @media (min-width: 1025px) {
+       .gradio-chatbot {
+           min-height: 500px;
+           max-height: calc(100vh - 200px);
+       }
+   }
+   
+   @media (max-height: 500px) and (orientation: landscape) {
+       .title {
+           font-size: 1.2rem;
+           margin-bottom: 0.25rem;
+       }
+       .gradio-chatbot {
+           min-height: 200px;
+           max-height: calc(100vh - 100px);
+       }
+   }
 """) as demo:
-    
-    with gr.Column(elem_classes=["main-container"]):
-        # Just the chatbot name at the top
-        gr.HTML("""
-        <div class='title'>🎓 EduBot</div>
-        """)
-        
-        # Flexible conversation viewing window
-        with gr.Column(elem_classes=["chat-container"]):
-            chatbot = gr.Chatbot(
-                show_copy_button=True,
-                avatar_images=[None, None],  # Remove avatars
-                placeholder="Hi! I'm EduBot. What would you like to learn today? 📚",
-                container=True,
-                elem_classes=["gradio-chatbot"]
-            )
-        
-        # Centered textbox and send button
-        with gr.Row(elem_classes=["input-row"]):
-            msg = gr.Textbox(
-                placeholder="Ask me about math, research, study strategies, or any educational topic...",
-                container=False,
-                show_label=False,
-                scale=8,
-                elem_classes=["custom-textbox"]
-            )
-            send_btn = gr.Button("📤", variant="primary", scale=1, elem_classes=["send-button"])
-    
-    # Enhanced response function
-    def respond_and_update(message, history):
-        for response in respond_with_enhanced_streaming(message, history):
-            history.append([message, response])
-            yield history, ""
-        
-    # Event handlers
-    msg.submit(
-        respond_and_update,
-        [msg, chatbot],
-        [chatbot, msg]
-    )
-    
-    send_btn.click(
-        respond_and_update,
-        [msg, chatbot],
-        [chatbot, msg]
-    )
+   
+   with gr.Column(elem_classes=["main-container"]):
+       gr.HTML("""
+       <div class='title'>🎓 EduBot</div>
+       """)
+       
+       with gr.Column(elem_classes=["chat-container"]):
+           chatbot = gr.Chatbot(
+               show_copy_button=True,
+               avatar_images=[None, None],
+               placeholder="Hi! I'm EduBot. What would you like to learn today? 📚",
+               container=True,
+               elem_classes=["gradio-chatbot"]
+           )
+       
+       with gr.Row(elem_classes=["input-row"]):
+           msg = gr.Textbox(
+               placeholder="Ask me about math, research, study strategies, or any educational topic...",
+               container=False,
+               show_label=False,
+               scale=8,
+               elem_classes=["custom-textbox"]
+           )
+           send_btn = gr.Button("📤", variant="primary", scale=1, elem_classes=["send-button"])
+   
+   def respond_and_update(message, history):
+       for response in respond_with_enhanced_streaming(message, history):
+           history.append([message, response])
+           yield history, ""
+       
+   msg.submit(
+       respond_and_update,
+       [msg, chatbot],
+       [chatbot, msg]
+   )
+   
+   send_btn.click(
+       respond_and_update,
+       [msg, chatbot],
+       [chatbot, msg]
+   )
 
 if __name__ == "__main__":
-    logger.info("Starting EduBot with custom styling...")
-    demo.launch()
+   logger.info("Starting EduBot with custom styling...")
+   demo.launch()