Created the interface

.gitignore

@@ -0,0 +1,2 @@
+venv/
+.env

app.py

@@ -1,7 +1,266 @@
 import gradio as gr
+import time
+from typing import Dict, List, Tuple, Optional
 
-def greet(name):
-    return "Hello " + name + "!!"
+# ==================== CORE SYSTEM STATE ====================
+class LawTrainingSystem:
+    def __init__(self):
+        self.session_active = False
+        self.current_case = None
+        self.student_side = None
+        self.conversation_history = []
+        self.processing_lock = False
+        
+    def initialize_session(self, case_title: str, case_description: str, student_side: str):
+        """Initialize a new training session"""
+        self.session_active = True
+        self.current_case = {"title": case_title, "description": case_description}
+        self.student_side = student_side
+        self.conversation_history = []
+        return f"Session initialized: {case_title} - You are the {student_side}"
+    
+    def is_processing(self) -> bool:
+        """Check if system is currently processing a response"""
+        return self.processing_lock
+    
+    def set_processing(self, status: bool):
+        """Set processing lock status"""
+        self.processing_lock = status
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Global system instance
+system = LawTrainingSystem()
+
+# ==================== AGENT RESPONSE SIMULATORS ====================
+def generate_judge_response(student_message: str) -> str:
+    """Simulate Judge AI response"""
+    # TODO: Replace with actual LangGraph Judge agent
+    time.sleep(1)  # Simulate processing time
+    return f"Judge: Your argument shows {['good', 'adequate', 'weak'][len(student_message) % 3]} legal reasoning. Consider strengthening your constitutional basis."
+
+def generate_opponent_response(student_message: str) -> str:
+    """Simulate Opponent AI response"""
+    # TODO: Replace with actual LangGraph Opponent agent
+    time.sleep(1.5)  # Simulate processing time
+    return f"Opponent: I object to your reasoning. The precedent clearly shows that {student_message[:20]}... is flawed because..."
+
+def generate_narrator_response(student_message: str) -> str:
+    """Simulate Narrator AI response"""
+    # TODO: Replace with actual LangGraph Narrator agent
+    time.sleep(0.8)  # Simulate processing time
+    return f"Narrator: The courtroom falls silent as the student presents their argument. The judge leans forward, considering the points raised..."
+
+# ==================== UI RESPONSE HANDLERS ====================
+def handle_student_response(message: str, judge_history: List, opponent_history: List, narrator_history: List) -> Tuple:
+    """Process student message and generate AI responses"""
+    
+    # Prevent multiple simultaneous processing
+    if system.is_processing():
+        return message, judge_history, opponent_history, narrator_history, gr.update(interactive=False)
+    
+    if not system.session_active:
+        return message, judge_history, opponent_history, narrator_history, gr.update(interactive=True)
+    
+    # Set processing lock
+    system.set_processing(True)
+    
+    try:
+        # Add student message to history
+        student_entry = [message, ""]
+        
+        # Generate AI responses (in sequence to avoid interference)
+        judge_response = generate_judge_response(message)
+        opponent_response = generate_opponent_response(message)
+        narrator_response = generate_narrator_response(message)
+        
+        # Update chat histories
+        judge_history.append(["", judge_response])
+        opponent_history.append(["", opponent_response])
+        narrator_history.append(["", narrator_response])
+        
+        # Clear student input and re-enable
+        return "", judge_history, opponent_history, narrator_history, gr.update(interactive=True)
+        
+    finally:
+        # Always release processing lock
+        system.set_processing(False)
+
+def start_case_session(case_title: str, case_description: str, student_side: str) -> Tuple:
+    """Initialize a new case session"""
+    if not case_title or not case_description:
+        return gr.update(), gr.update(), gr.update(), gr.update(), gr.update(interactive=False)
+    
+    # Initialize system
+    init_message = system.initialize_session(case_title, case_description, student_side)
+    
+    # Generate initial context from Narrator
+    initial_context = generate_narrator_response(f"Case: {case_title}")
+    
+    # Create initial chat histories
+    judge_history = [["", "Judge: I am ready to evaluate your arguments based on legal precedent and constitutional law."]]
+    opponent_history = [["", f"Opponent: I will argue for the {('prosecution' if student_side == 'defense' else 'defense')} side."]]
+    narrator_history = [["", initial_context]]
+    
+    return (
+        judge_history,
+        opponent_history,
+        narrator_history,
+        gr.update(visible=True),
+        gr.update(interactive=True)
+    )
+
+def upload_case_documents(files) -> str:
+    """Handle case document uploads"""
+    if not files:
+        return "No documents uploaded"
+    
+    # TODO: Process uploaded documents with document parsing
+    file_names = [f.name for f in files]
+    return f"Uploaded {len(files)} documents: {', '.join(file_names)}"
+
+# ==================== GRADIO INTERFACE ====================
+def create_law_training_interface():
+    """Create the main Gradio interface"""
+    
+    with gr.Blocks(
+        title="Law Training Agentic System",
+        theme=gr.themes.Soft(),
+        css="""
+        .chat-container { height: 400px; }
+        .upload-section { border: 2px dashed #ccc; padding: 20px; margin: 10px 0; }
+        .character-panel { border: 1px solid #ddd; border-radius: 8px; margin: 5px; }
+        .student-panel { border: 2px solid #4CAF50; }
+        """
+    ) as interface:
+        
+        # Header
+        gr.Markdown("# ⚖️ Law Training Agentic System")
+        gr.Markdown("Practice your courtroom argumentation skills with AI-powered opponents, judges, and narrators")
+        
+        # ==================== CASE SETUP SECTION ====================
+        with gr.Row():
+            with gr.Column(scale=2):
+                gr.Markdown("## 📋 Case Setup")
+                case_title = gr.Textbox(
+                    label="Case Title",
+                    placeholder="e.g., Freedom of Speech vs. Public Safety",
+                    lines=1
+                )
+                case_description = gr.Textbox(
+                    label="Case Description",
+                    placeholder="Describe the legal scenario you want to practice...",
+                    lines=3
+                )
+                student_side = gr.Radio(
+                    choices=["prosecution", "defense"],
+                    label="Your Role",
+                    value="prosecution"
+                )
+                
+            with gr.Column(scale=1):
+                gr.Markdown("## 📁 Documents")
+                file_upload = gr.File(
+                    label="Upload Case Documents",
+                    file_count="multiple",
+                    file_types=[".pdf", ".txt", ".docx"],
+                    elem_classes="upload-section"
+                )
+                upload_status = gr.Textbox(
+                    label="Upload Status",
+                    interactive=False,
+                    lines=2
+                )
+        
+        # Start session button
+        start_btn = gr.Button("🚀 Start Training Session", variant="primary", size="lg")
+        
+        # ==================== COURTROOM INTERFACE ====================
+        courtroom_section = gr.Group(visible=False)
+        
+        with courtroom_section:
+            gr.Markdown("## 🏛️ Courtroom")
+            
+            # Four-panel layout for characters
+            with gr.Row(equal_height=True):
+                # Student Panel (Interactive)
+                with gr.Column(scale=1, elem_classes="character-panel student-panel"):
+                    gr.Markdown("### 🎓 **Student (You)**")
+                    student_input = gr.Textbox(
+                        label="Your Argument",
+                        placeholder="Present your legal argument here...",
+                        lines=3,
+                        interactive=False
+                    )
+                    submit_btn = gr.Button("📝 Submit Argument", variant="primary")
+                
+                # Judge Panel (AI Response Only)
+                with gr.Column(scale=1, elem_classes="character-panel"):
+                    gr.Markdown("### ⚖️ **Judge**")
+                    judge_chat = gr.Chatbot(
+                        label="Judge Feedback",
+                        elem_classes="chat-container",
+                        show_label=False
+                    )
+            
+            with gr.Row(equal_height=True):
+                # Opponent Panel (AI Response Only)
+                with gr.Column(scale=1, elem_classes="character-panel"):
+                    gr.Markdown("### 🥊 **Opponent**")
+                    opponent_chat = gr.Chatbot(
+                        label="Opponent Arguments",
+                        elem_classes="chat-container",
+                        show_label=False
+                    )
+                
+                # Narrator Panel (AI Response Only)
+                with gr.Column(scale=1, elem_classes="character-panel"):
+                    gr.Markdown("### 📖 **Narrator**")
+                    narrator_chat = gr.Chatbot(
+                        label="Courtroom Narration",
+                        elem_classes="chat-container",
+                        show_label=False
+                    )
+        
+        # ==================== EVENT HANDLERS ====================
+        
+        # File upload handler
+        file_upload.change(
+            fn=upload_case_documents,
+            inputs=[file_upload],
+            outputs=[upload_status]
+        )
+        
+        # Start session handler
+        start_btn.click(
+            fn=start_case_session,
+            inputs=[case_title, case_description, student_side],
+            outputs=[judge_chat, opponent_chat, narrator_chat, courtroom_section, student_input]
+        )
+        
+        # Student response handler
+        submit_btn.click(
+            fn=handle_student_response,
+            inputs=[student_input, judge_chat, opponent_chat, narrator_chat],
+            outputs=[student_input, judge_chat, opponent_chat, narrator_chat, submit_btn]
+        )
+        
+        # Enter key submission
+        student_input.submit(
+            fn=handle_student_response,
+            inputs=[student_input, judge_chat, opponent_chat, narrator_chat],
+            outputs=[student_input, judge_chat, opponent_chat, narrator_chat, submit_btn]
+        )
+    
+    return interface
+
+# ==================== MAIN EXECUTION ====================
+if __name__ == "__main__":
+    # Create and launch the interface
+    app = create_law_training_interface()
+    
+    # Launch configuration
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        debug=True
+    )