Update app.py

app.py

@@ -288,6 +288,67 @@ class PowerSystemsConsultant:
         
         return None
     
+    def generate_practice_pack(self, topic: str, difficulty: str, num_questions: int) -> str:
+        """Generate practice questions pack"""
+        try:
+            practice_prompt = f"""Generate {num_questions} power systems practice questions about {topic} at {difficulty} level.
+            Format the response with:
+            1. Question number and statement
+            2. Multiple choice options (A, B, C, D) when appropriate
+            3. Detailed solution with step-by-step calculations
+            4. Key concepts and formulas used
+            5. References to relevant standards
+            
+            Focus on practical engineering scenarios and include numerical calculations where applicable."""
+            
+            messages = [
+                {"role": "system", "content": "You are an expert power systems engineer creating practice questions for students and professionals."},
+                {"role": "user", "content": practice_prompt}
+            ]
+            
+            response = self.groq_client.chat.completions.create(
+                model="llama3-70b-8192",
+                messages=messages,
+                max_tokens=3000,
+                temperature=0.7
+            )
+            
+            return response.choices[0].message.content
+            
+        except Exception as e:
+            return f"Error generating practice pack: {str(e)}"
+    
+    def explain_standard(self, standard: str) -> str:
+        """Explain power systems standard"""
+        try:
+            standard_prompt = f"""Provide a comprehensive explanation of the {standard} standard including:
+            1. Purpose and scope of the standard
+            2. Key requirements and specifications
+            3. Practical applications in power systems
+            4. Important clauses and sections
+            5. Relationship to other standards
+            6. Implementation considerations
+            7. Common misconceptions or challenges
+            
+            Make it practical and useful for power systems engineers."""
+            
+            messages = [
+                {"role": "system", "content": "You are an expert in power systems standards and regulations with deep knowledge of IEEE, IEC, and ANSI standards."},
+                {"role": "user", "content": standard_prompt}
+            ]
+            
+            response = self.groq_client.chat.completions.create(
+                model="llama3-70b-8192",
+                messages=messages,
+                max_tokens=2500,
+                temperature=0.6
+            )
+            
+            return response.choices[0].message.content
+            
+        except Exception as e:
+            return f"Error explaining standard: {str(e)}"
+    
     def create_chat_session(self, user_id: int, title: str = None) -> str:
         """Create new chat session"""
         session_id = str(uuid.uuid4())
@@ -772,6 +833,9 @@ def create_app():
                                 
                                 <h4>Line-to-Line Fault:</h4>
                                 <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>I_fault = √3 × V_nominal / (Z1 + Z2)</code></p>
+                                
+                                <h4>Double Line-to-Ground Fault:</h4>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>I_fault = V_nominal × [Z2 + Z0] / [Z1 × (Z2 + Z0) + Z2 × Z0]</code></p>
                             </div>
                         """)
                     
@@ -782,4 +846,240 @@ def create_app():
                                 <ul style='margin-left: 1.5rem;'>
                                     <li>Inverse Time Characteristics</li>
                                     <li>Coordination Time Intervals (0.2-0.5s)</li>
-                                    <li>Pickup Current Settings (1.25-1.5
+                                    <li>Pickup Current Settings (1.25-1.5 × Full Load)</li>
+                                    <li>Time Dial Settings</li>
+                                </ul>
+                                
+                                <h4>Distance Protection:</h4>
+                                <ul style='margin-left: 1.5rem;'>
+                                    <li>Zone 1: 80-85% of line length</li>
+                                    <li>Zone 2: 120% of protected line + 50% of next</li>
+                                    <li>Zone 3: Backup protection</li>
+                                </ul>
+                                
+                                <h4>Differential Protection:</h4>
+                                <ul style='margin-left: 1.5rem;'>
+                                    <li>CT ratio matching</li>
+                                    <li>Restraint characteristics</li>
+                                    <li>Second harmonic blocking</li>
+                                </ul>
+                            </div>
+                        """)
+                    
+                    with gr.Accordion("📐 Power System Calculations", open=False):
+                        gr.HTML("""
+                            <div style='background: #f8f9fa; padding: 1.5rem; border-radius: 8px; margin: 1rem 0;'>
+                                <h4>Power Formulas:</h4>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>P = √3 × V × I × cos(φ)</code> (Three-phase real power)</p>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>Q = √3 × V × I × sin(φ)</code> (Three-phase reactive power)</p>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>S = √3 × V × I</code> (Three-phase apparent power)</p>
+                                
+                                <h4>Impedance Calculations:</h4>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>Z_pu = Z_actual × (S_base / V_base²)</code></p>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>Z_new = Z_old × (S_new/S_old) × (V_old/V_new)²</code></p>
+                                
+                                <h4>Transformer Calculations:</h4>
+                                <p><code style='background: #e9ecef; padding: 0.25rem 0.5rem; border-radius: 4px;'>X_pu = (% impedance / 100) × (S_base / S_transformer)</code></p>
+                            </div>
+                        """)
+                    
+                    with gr.Accordion("📊 Standards Quick Reference", open=False):
+                        gr.HTML("""
+                            <div style='background: #f8f9fa; padding: 1.5rem; border-radius: 8px; margin: 1rem 0;'>
+                                <h4>IEEE Standards:</h4>
+                                <ul style='margin-left: 1.5rem;'>
+                                    <li><strong>C37.2:</strong> Device Function Numbers</li>
+                                    <li><strong>519:</strong> Harmonic Control</li>
+                                    <li><strong>1547:</strong> Distributed Generation Interconnection</li>
+                                    <li><strong>242:</strong> Protection and Coordination</li>
+                                    <li><strong>399:</strong> Industrial Power Systems Analysis</li>
+                                </ul>
+                                
+                                <h4>IEC Standards:</h4>
+                                <ul style='margin-left: 1.5rem;'>
+                                    <li><strong>61850:</strong> Substation Automation</li>
+                                    <li><strong>60909:</strong> Short-Circuit Calculations</li>
+                                    <li><strong>60255:</strong> Electrical Relays</li>
+                                    <li><strong>61131:</strong> PLC Programming</li>
+                                </ul>
+                            </div>
+                        """)
+                    
+                    with gr.Accordion("🎓 Exam Preparation Tips", open=False):
+                        gr.HTML("""
+                            <div style='background: #f8f9fa; padding: 1.5rem; border-radius: 8px; margin: 1rem 0;'>
+                                <h4>PE Exam Power Topics:</h4>
+                                <ul style='margin-left: 1.5rem;'>
+                                    <li>Three-phase circuit analysis</li>
+                                    <li>Symmetrical components</li>
+                                    <li>Fault analysis calculations</li>
+                                    <li>Protection system coordination</li>
+                                    <li>Power flow studies</li>
+                                    <li>Economic dispatch</li>
+                                    <li>Stability studies</li>
+                                    <li>Grounding systems</li>
+                                </ul>
+                                
+                                <h4>Key Formulas to Memorize:</h4>
+                                <ul style='margin-left: 1.5rem;'>
+                                    <li>Symmetrical component transformations</li>
+                                    <li>Fault current calculations</li>
+                                    <li>Power triangle relationships</li>
+                                    <li>Per-unit system conversions</li>
+                                    <li>Transmission line parameters</li>
+                                </ul>
+                            </div>
+                        """)
+        
+        # Event handlers for login
+        def handle_login(username, password):
+            success, user_data, message, sessions_update, main_visibility = login_user(username, password)
+            if success:
+                welcome_html = f"""
+                    <div class="feature-box" style="text-align: center;">
+                        <h2>👋 Welcome, {user_data['full_name']}!</h2>
+                        <p>Ready to tackle power systems challenges? Let's get started!</p>
+                    </div>
+                """
+                return (
+                    message,
+                    user_data,
+                    gr.update(visible=False),  # Hide login
+                    gr.update(visible=True),   # Show main
+                    welcome_html,
+                    sessions_update
+                )
+            else:
+                return (
+                    message,
+                    None,
+                    gr.update(visible=True),   # Keep login visible
+                    gr.update(visible=False),  # Keep main hidden
+                    "",
+                    gr.update()
+                )
+        
+        def handle_registration(username, email, password, confirm_password, full_name):
+            success, message = register_user(username, email, password, confirm_password, full_name)
+            return message
+        
+        def handle_new_chat(user_data):
+            if user_data:
+                session_id, session_choices = start_new_chat(user_data)
+                session_choices_formatted = [(f"{s['title']} ({s['created_at'][:10]})", s['session_id']) 
+                                           for s in consultant.user_manager.get_user_sessions(user_data['id'])]
+                return (
+                    session_id,
+                    gr.update(choices=session_choices_formatted, value=session_id),
+                    []  # Clear chat history
+                )
+            return None, gr.update(), []
+        
+        def handle_chat_selection(session_id, user_data):
+            if session_id and user_data:
+                history = load_selected_chat(session_id, user_data)
+                return history, session_id
+            return [], None
+        
+        def handle_chat_message(message, history, user_data, session_id):
+            return chat_interface(message, history, user_data, session_id)
+        
+        def clear_chat():
+            return [], ""
+        
+        def handle_practice_generation(topic, difficulty, num_questions, user_data):
+            return generate_practice_questions(topic, difficulty, num_questions, user_data)
+        
+        def handle_standard_explanation(standard, user_data):
+            return explain_standard_interface(standard, user_data)
+        
+        # Wire up event handlers
+        login_btn.click(
+            fn=handle_login,
+            inputs=[login_username, login_password],
+            outputs=[login_status, user_state, login_section, main_section, user_welcome, chat_sessions_dropdown]
+        )
+        
+        register_btn.click(
+            fn=handle_registration,
+            inputs=[reg_username, reg_email, reg_password, reg_confirm_password, reg_full_name],
+            outputs=[register_status]
+        )
+        
+        new_chat_btn.click(
+            fn=handle_new_chat,
+            inputs=[user_state],
+            outputs=[session_state, chat_sessions_dropdown, chatbot]
+        )
+        
+        chat_sessions_dropdown.change(
+            fn=handle_chat_selection,
+            inputs=[chat_sessions_dropdown, user_state],
+            outputs=[chatbot, session_state]
+        )
+        
+        submit_btn.click(
+            fn=handle_chat_message,
+            inputs=[msg, chatbot, user_state, session_state],
+            outputs=[chatbot, msg, diagram_display]
+        )
+        
+        msg.submit(
+            fn=handle_chat_message,
+            inputs=[msg, chatbot, user_state, session_state],
+            outputs=[chatbot, msg, diagram_display]
+        )
+        
+        clear_btn.click(
+            fn=clear_chat,
+            outputs=[chatbot, msg]
+        )
+        
+        generate_btn.click(
+            fn=handle_practice_generation,
+            inputs=[topic_input, difficulty_input, num_questions_input, user_state],
+            outputs=[practice_output]
+        )
+        
+        explain_btn.click(
+            fn=handle_standard_explanation,
+            inputs=[standard_input, user_state],
+            outputs=[standard_output]
+        )
+    
+    return app
+
+# Launch the application
+if __name__ == "__main__":
+    app = create_app()
+    
+    # Add some helpful startup information
+    print("🚀 Power Systems Mini-Consultant Starting...")
+    print("📋 Features included:")
+    print("   - AI-powered chat consultant")
+    print("   - Practice question generator")
+    print("   - Standards explanations")
+    print("   - Engineering diagram generation")
+    print("   - User authentication & session management")
+    print("   - Chat history persistence")
+    print()
+    
+    if consultant is None:
+        print("⚠️  Warning: System not fully initialized")
+        print("   Please ensure GROQ_API_KEY is set in your environment")
+        print("   Some features may not work properly")
+    else:
+        print("✅ System ready!")
+    
+    print()
+    print("🌐 Access your consultant at: http://localhost:7860")
+    print("📚 Study resources and formulas available in the Resources tab")
+    print("💡 Try asking about fault analysis, protection systems, or request diagrams!")
+    
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_api=False,
+        quiet=False
+    )