| from fastapi import FastAPI, WebSocket |
| from fastapi.responses import HTMLResponse |
| import os |
|
|
|
|
| app = FastAPI() |
|
|
| html = """ |
| <!DOCTYPE html> |
| <html> |
| <head> |
| <title>Chat</title> |
| </head> |
| <body> |
| <h1>WebSocket Chat</h1> |
| <form action="" onsubmit="sendMessage(event)"> |
| <input type="text" id="messageText" autocomplete="off"/> |
| <button>Send</button> |
| </form> |
| <ul id='messages'> |
| </ul> |
| <script> |
| // var ws = new WebSocket("ws://localhost:8000/ws"); |
| var ws = new WebSocket("wss://clone3-websocket.hf.space/ws"); |
| ws.onmessage = function(event) { |
| var messages = document.getElementById('messages') |
| var message = document.createElement('li') |
| var content = document.createTextNode(event.data) |
| message.appendChild(content) |
| messages.appendChild(message) |
| }; |
| function sendMessage(event) { |
| var input = document.getElementById("messageText") |
| ws.send(input.value) |
| input.value = '' |
| event.preventDefault() |
| } |
| </script> |
| </body> |
| </html> |
| """ |
|
|
|
|
| @app.get("/") |
| async def get(): |
| return HTMLResponse(html) |
|
|
| @app.get("/env") |
| async def env(): |
| environment_variables = "<h3>Environment Variables</h3>" |
| for name, value in os.environ.items(): |
| environment_variables += f"{name}: {value}<br>" |
| return HTMLResponse(environment_variables) |
|
|
| @app.websocket("/ws") |
| async def websocket_endpoint(websocket: WebSocket): |
| await websocket.accept() |
| while True: |
| data = await websocket.receive_text() |
| await websocket.send_text(f"Message text was: {data}") |
|
|
|
|
