Check: fix the input() not working in normal scripts

Author: Sachin1801

deployment/get-staging-url.sh

@@ -0,0 +1,74 @@
+#!/bin/bash
+# Quick script to find the current staging environment URL
+# ECS tasks with public IPs get new IPs on every redeployment
+
+set -e
+
+echo "🔍 Finding staging environment URL..."
+echo ""
+
+# Ensure AWS profile is set
+export AWS_PROFILE=personal
+REGION="us-east-2"
+CLUSTER="pythonide-cluster"
+SERVICE="pythonide-staging-service"
+
+# Get the running task ARN
+echo "📋 Getting task ARN..."
+TASK_ARN=$(aws ecs list-tasks \
+    --cluster "$CLUSTER" \
+    --service-name "$SERVICE" \
+    --region "$REGION" \
+    --query 'taskArns[0]' \
+    --output text)
+
+if [ -z "$TASK_ARN" ] || [ "$TASK_ARN" = "None" ]; then
+    echo "❌ No running tasks found for staging service"
+    exit 1
+fi
+
+TASK_ID=$(echo "$TASK_ARN" | awk -F'/' '{print $NF}')
+echo "✅ Task ID: $TASK_ID"
+echo ""
+
+# Get network interface ID from task
+echo "🌐 Getting network interface..."
+ENI=$(aws ecs describe-tasks \
+    --cluster "$CLUSTER" \
+    --tasks "$TASK_ID" \
+    --region "$REGION" \
+    --query 'tasks[0].attachments[0].details[?name==`networkInterfaceId`].value' \
+    --output text)
+
+if [ -z "$ENI" ] || [ "$ENI" = "None" ]; then
+    echo "❌ Could not find network interface for task"
+    exit 1
+fi
+
+echo "✅ Network Interface: $ENI"
+echo ""
+
+# Get public IP from network interface
+echo "📍 Getting public IP..."
+PUBLIC_IP=$(aws ec2 describe-network-interfaces \
+    --network-interface-ids "$ENI" \
+    --region "$REGION" \
+    --query 'NetworkInterfaces[0].Association.PublicIp' \
+    --output text)
+
+if [ -z "$PUBLIC_IP" ] || [ "$PUBLIC_IP" = "None" ]; then
+    echo "❌ Task does not have a public IP"
+    exit 1
+fi
+
+echo "✅ Public IP: $PUBLIC_IP"
+echo ""
+echo "================================================"
+echo "🎯 STAGING URL:"
+echo "   http://$PUBLIC_IP:8080"
+echo "================================================"
+echo ""
+echo "📝 Credentials:"
+echo "   Admin:   sa9082 / Admin@sa9082"
+echo "   Student: test_1 / student@test_1"
+echo ""

server/command/hybrid_repl_thread.py

@@ -987,8 +987,47 @@ def read_pty_output(self):
                         print(f"[PTY-DEBUG] ✓ Timeout monitor will now exit (REPL mode active)")
                         self.response_to_client(5000, {"mode": "repl"})
 
-                    # For PTY mode with native input(), we don't need INPUT_REQUEST markers
-                    # The prompts appear naturally in the output stream
+                    # Check for matplotlib figures
+                    if "__MATPLOTLIB_FIGURE_START__" in buffer and "__MATPLOTLIB_FIGURE_END__" in buffer:
+                        start = buffer.index("__MATPLOTLIB_FIGURE_START__") + len("__MATPLOTLIB_FIGURE_START__")
+                        end = buffer.index("__MATPLOTLIB_FIGURE_END__")
+                        figure_data = buffer[start:end].strip()
+
+                        # Send figure to client
+                        self.response_to_client(3000, {"figure": figure_data})
+
+                        # Clear the marker from buffer
+                        buffer = buffer[end + len("__MATPLOTLIB_FIGURE_END__"):]
+
+                    # CRITICAL: Check for INPUT_REQUEST markers from custom input() during script execution
+                    if "__INPUT_REQUEST_START__" in buffer and "__INPUT_REQUEST_END__" in buffer:
+                        # Extract the positions
+                        marker_start = buffer.index("__INPUT_REQUEST_START__")
+                        marker_end = buffer.index("__INPUT_REQUEST_END__") + len("__INPUT_REQUEST_END__")
+
+                        # Extract prompt from markers
+                        prompt_start = marker_start + len("__INPUT_REQUEST_START__")
+                        prompt_end = buffer.index("__INPUT_REQUEST_END__")
+                        prompt = buffer[prompt_start:prompt_end]
+
+                        # Send any text before the marker
+                        pre_marker = buffer[:marker_start]
+                        if pre_marker and pre_marker.strip():
+                            for line in pre_marker.split("\n"):
+                                if line or line == "":
+                                    self.response_to_client(0, {"stdout": line})
+
+                        print(f"[PTY-DEBUG] Input request detected: {repr(prompt)}")
+
+                        # Pause timeout when waiting for user input
+                        self.waiting_for_input = True
+
+                        # Send input request signal to enable input field (with prompt for frontend)
+                        # The frontend will display the prompt, so we don't need to send it separately
+                        self.response_to_client(2000, {"prompt": prompt})
+
+                        # Clear the processed part from buffer
+                        buffer = buffer[marker_end:].lstrip('\n')
 
                     # Send complete lines to client
                     while '\n' in buffer or '\r' in buffer: