Update lm_trial_app.py

lm_trial_app.py

@@ -86,6 +86,16 @@ Format your response as:
 PROBLEM: [The problem description and requirements]
 CODE: [Any starter code if applicable, or "Write your code here:"]
 
+IMPORTANT: If you include example outputs, format them as code blocks like this:
+```
+Example Output:
+1
+1 2
+1 2 3
+1 2 3 4
+1 2 3 4 5
+```
+
 Keep it concise but clear."""
         else:
             # Use curriculum-based prompt
@@ -110,6 +120,16 @@ Format your response as:
 PROBLEM: [The problem description and requirements]
 CODE: [Any starter code if applicable, or "Write your code here:"]
 
+IMPORTANT: If you include example outputs, format them as code blocks like this:
+```
+Example Output:
+1
+1 2
+1 2 3
+1 2 3 4
+1 2 3 4 5
+```
+
 Keep it concise but clear and aligned with the curriculum difficulty."""
 
         try:
@@ -244,6 +264,40 @@ Be encouraging but honest. Focus on learning and improvement based on the curric
         except Exception as e:
             return f"Error analyzing code: {str(e)}"
 
+    def execute_code(self, code):
+        """Execute the student's code and return the output"""
+        try:
+            import subprocess
+            import sys
+            from io import StringIO
+            import contextlib
+            
+            # Create a safe execution environment
+            local_vars = {}
+            output_buffer = StringIO()
+            
+            # Capture stdout and stderr
+            with contextlib.redirect_stdout(output_buffer), contextlib.redirect_stderr(output_buffer):
+                try:
+                    # Execute the code
+                    exec(code, {"__builtins__": __builtins__}, local_vars)
+                    output = output_buffer.getvalue()
+                    
+                    if output.strip():
+                        return f"$ python code.py\n{output}"
+                    else:
+                        return f"$ python code.py\n# Code executed successfully but no output produced"
+                        
+                except Exception as e:
+                    error_output = output_buffer.getvalue()
+                    if error_output.strip():
+                        return f"$ python code.py\n{error_output}\nError: {str(e)}"
+                    else:
+                        return f"$ python code.py\nError: {str(e)}"
+                    
+        except Exception as e:
+            return f"$ python code.py\nError: Could not execute code - {str(e)}"
+
 # --- LLM-Powered Curriculum Assistant ---
 
 class LLMCurriculumAssistant:
@@ -534,9 +588,20 @@ def generate_problem(topic, problem_type):
     return problem, code
 
 def analyze_code(topic, problem_type, problem_description, student_code):
-    """Analyze student's code"""
-    analysis = practice_assistant.analyze_student_code(topic, problem_type, problem_description, student_code)
-    return analysis
+    """Analyze student's code and execute it"""
+    try:
+        # Execute the code to get terminal output
+        terminal_output = practice_assistant.execute_code(student_code)
+    except Exception as e:
+        terminal_output = f"$ python code.py\nError: Could not execute code - {str(e)}"
+    
+    try:
+        # Get LLM analysis (this should always run regardless of execution errors)
+        analysis = practice_assistant.analyze_student_code(topic, problem_type, problem_description, student_code)
+    except Exception as e:
+        analysis = f"Error getting LLM analysis: {str(e)}"
+    
+    return terminal_output, analysis
 
 with gr.Blocks(title="LLM Curriculum Assistant", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🤖 LLM Curriculum Assistant\nYour AI programming tutor with LLM-powered content selection and code practice!")
@@ -628,6 +693,14 @@ with gr.Blocks(title="LLM Curriculum Assistant", theme=gr.themes.Soft()) as demo
                     
                     analyze_btn = gr.Button("🔍 Analyze My Code", variant="secondary", size="lg")
                     
+                    gr.Markdown("#### 💻 Code Execution Output")
+                    terminal_output = gr.Code(
+                        label="Terminal Output",
+                        language="bash",
+                        lines=8,
+                        value="# Code execution output will appear here..."
+                    )
+                    
                     gr.Markdown("#### 📊 AI Analysis")
                     analysis_output = gr.Markdown(label="Analysis will appear here...")
             
@@ -641,7 +714,7 @@ with gr.Blocks(title="LLM Curriculum Assistant", theme=gr.themes.Soft()) as demo
             analyze_btn.click(
                 fn=analyze_code,
                 inputs=[topic_input, problem_type, problem_description, student_code],
-                outputs=[analysis_output]
+                outputs=[terminal_output, analysis_output]
             )
 
 if __name__ == "__main__":