mirror of
https://github.com/HumanAIGC-Engineering/gradio-webrtc.git
synced 2026-02-04 17:39:23 +08:00
aefb08150f
sync code of fastrtc, add text support through datachannel, fix safari connect problem support chat without camera or mic
46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
from fastapi import FastAPI
|
|
from fastapi.responses import RedirectResponse
|
|
from fastrtc import Stream
|
|
from gradio.utils import get_space
|
|
|
|
try:
|
|
from demo.llama_code_editor.handler import (
|
|
CodeHandler,
|
|
)
|
|
from demo.llama_code_editor.ui import demo as ui
|
|
except (ImportError, ModuleNotFoundError):
|
|
from handler import CodeHandler
|
|
from ui import demo as ui
|
|
|
|
|
|
stream = Stream(
|
|
handler=CodeHandler,
|
|
modality="audio",
|
|
mode="send-receive",
|
|
concurrency_limit=10 if get_space() else None,
|
|
time_limit=90 if get_space() else None,
|
|
)
|
|
|
|
stream.ui = ui
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
@app.get("/")
|
|
async def _():
|
|
url = "/ui" if not get_space() else "https://fastrtc-llama-code-editor.hf.space/ui/"
|
|
return RedirectResponse(url)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import os
|
|
|
|
if (mode := os.getenv("MODE")) == "UI":
|
|
stream.ui.launch(server_port=7860, server_name="0.0.0.0")
|
|
elif mode == "PHONE":
|
|
stream.fastphone(host="0.0.0.0", port=7860)
|
|
else:
|
|
import uvicorn
|
|
|
|
uvicorn.run(app, host="0.0.0.0", port=7860)
|