Spaces:
Running
Running
File size: 3,554 Bytes
0d9f1af efe2c71 0d9f1af 24f0634 0d9f1af 5eb1bc0 0d9f1af 5eb1bc0 efe2c71 5eb1bc0 0d9f1af 5eb1bc0 efe2c71 0d9f1af efe2c71 0d9f1af efe2c71 0d9f1af efe2c71 0d9f1af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
import React, { useState } from "react";
import { Button } from "../components/ui/button";
import {
Card,
CardContent,
CardDescription,
CardHeader,
CardTitle,
} from "../components/ui/card";
import { Alert, AlertDescription } from "../components/ui/alert";
import { PortManager } from "../components/PortManager";
import { CalibrationPanel } from "../components/CalibrationPanel";
import { TeleoperationPanel } from "../components/TeleoperationPanel";
import { isWebSerialSupported } from "../../lerobot/web/calibrate";
import type { RobotConnection } from "../../lerobot/web/types/robot-connection.js";
interface HomeProps {
onGetStarted: () => void;
connectedRobots: RobotConnection[];
onConnectedRobotsChange: (robots: RobotConnection[]) => void;
}
export function Home({
onGetStarted,
connectedRobots,
onConnectedRobotsChange,
}: HomeProps) {
const [calibratingRobot, setCalibratingRobot] =
useState<RobotConnection | null>(null);
const [teleoperatingRobot, setTeleoperatingRobot] =
useState<RobotConnection | null>(null);
const isSupported = isWebSerialSupported();
const handleCalibrate = (
port: SerialPort,
robotType: "so100_follower" | "so100_leader",
robotId: string
) => {
// Find the robot from connectedRobots
const robot = connectedRobots.find((r) => r.port === port);
if (robot) {
setCalibratingRobot(robot);
}
};
const handleTeleoperate = (robot: RobotConnection) => {
setTeleoperatingRobot(robot);
};
const handleFinishCalibration = () => {
setCalibratingRobot(null);
};
const handleFinishTeleoperation = () => {
setTeleoperatingRobot(null);
};
return (
<div className="min-h-screen bg-gradient-to-br from-blue-50 to-indigo-100">
<div className="container mx-auto px-6 py-12">
{/* Header */}
<div className="text-center mb-12">
<h1 className="text-4xl font-bold text-gray-900 mb-4">
🤖 LeRobot.js
</h1>
<p className="text-xl text-gray-600 mb-8">
State-of-the-art AI for real-world robotics in JavaScript
</p>
{!isSupported && (
<Alert variant="destructive" className="max-w-2xl mx-auto mb-8">
<AlertDescription>
Web Serial API is not supported in this browser. Please use
Chrome, Edge, or another Chromium-based browser to use this
demo.
</AlertDescription>
</Alert>
)}
</div>
{/* Main Content */}
{calibratingRobot ? (
<div className="max-w-6xl mx-auto">
<div className="mb-4">
<Button
variant="outline"
onClick={() => setCalibratingRobot(null)}
>
← Back to Dashboard
</Button>
</div>
<CalibrationPanel
robot={calibratingRobot}
onFinish={handleFinishCalibration}
/>
</div>
) : teleoperatingRobot ? (
<TeleoperationPanel
robot={teleoperatingRobot}
onClose={handleFinishTeleoperation}
/>
) : (
<div className="max-w-6xl mx-auto">
<PortManager
onCalibrate={handleCalibrate}
onTeleoperate={handleTeleoperate}
connectedRobots={connectedRobots}
onConnectedRobotsChange={onConnectedRobotsChange}
/>
</div>
)}
</div>
</div>
);
}
|