freddyaboulton HF staff commited on
Commit
e0058ac
·
verified ·
1 Parent(s): f6e16c3

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. app.py +14 -4
  2. handler.py +9 -8
  3. ui.py +5 -4
app.py CHANGED
@@ -1,5 +1,6 @@
1
- from fastrtc import Stream
2
  from fastapi.responses import RedirectResponse
 
3
  from gradio.utils import get_space
4
 
5
  try:
@@ -22,14 +23,23 @@ stream = Stream(
22
 
23
  stream.ui = ui
24
 
 
25
 
26
- @stream.get("/")
 
27
  async def _():
28
  url = "/ui" if not get_space() else "https://fastrtc-llama-code-editor.hf.space/ui/"
29
  return RedirectResponse(url)
30
 
31
 
32
  if __name__ == "__main__":
33
- import uvicorn
 
 
 
 
 
 
 
34
 
35
- uvicorn.run(stream, host="0.0.0.0", port=7860)
 
1
+ from fastapi import FastAPI
2
  from fastapi.responses import RedirectResponse
3
+ from fastrtc import Stream
4
  from gradio.utils import get_space
5
 
6
  try:
 
23
 
24
  stream.ui = ui
25
 
26
+ app = FastAPI()
27
 
28
+
29
+ @app.get("/")
30
  async def _():
31
  url = "/ui" if not get_space() else "https://fastrtc-llama-code-editor.hf.space/ui/"
32
  return RedirectResponse(url)
33
 
34
 
35
  if __name__ == "__main__":
36
+ import os
37
+
38
+ if (mode := os.getenv("MODE")) == "UI":
39
+ stream.ui.launch(server_port=7860, server_name="0.0.0.0")
40
+ elif mode == "PHONE":
41
+ stream.fastphone(host="0.0.0.0", port=7860)
42
+ else:
43
+ import uvicorn
44
 
45
+ uvicorn.run(app, host="0.0.0.0", port=7860)
handler.py CHANGED
@@ -1,16 +1,17 @@
 
 
 
 
 
 
 
 
1
  from fastrtc import (
2
- ReplyOnPause,
3
  AdditionalOutputs,
 
4
  audio_to_bytes,
5
  )
6
- import numpy as np
7
- import os
8
- import base64
9
- import openai
10
- import re
11
  from groq import Groq
12
- from dotenv import load_dotenv
13
- from pathlib import Path
14
 
15
  load_dotenv()
16
 
 
1
+ import base64
2
+ import os
3
+ import re
4
+ from pathlib import Path
5
+
6
+ import numpy as np
7
+ import openai
8
+ from dotenv import load_dotenv
9
  from fastrtc import (
 
10
  AdditionalOutputs,
11
+ ReplyOnPause,
12
  audio_to_bytes,
13
  )
 
 
 
 
 
14
  from groq import Groq
 
 
15
 
16
  load_dotenv()
17
 
ui.py CHANGED
@@ -1,17 +1,18 @@
 
 
 
1
  from dotenv import load_dotenv
2
  from fastrtc import WebRTC, get_twilio_turn_credentials
3
- import gradio as gr
4
  from gradio.utils import get_space
5
- from pathlib import Path
6
 
7
  try:
8
  from demo.llama_code_editor.handler import (
9
  CodeHandler,
10
- system_prompt,
11
  display_in_sandbox,
 
12
  )
13
  except (ImportError, ModuleNotFoundError):
14
- from handler import CodeHandler, system_prompt, display_in_sandbox
15
 
16
  load_dotenv()
17
 
 
1
+ from pathlib import Path
2
+
3
+ import gradio as gr
4
  from dotenv import load_dotenv
5
  from fastrtc import WebRTC, get_twilio_turn_credentials
 
6
  from gradio.utils import get_space
 
7
 
8
  try:
9
  from demo.llama_code_editor.handler import (
10
  CodeHandler,
 
11
  display_in_sandbox,
12
+ system_prompt,
13
  )
14
  except (ImportError, ModuleNotFoundError):
15
+ from handler import CodeHandler, display_in_sandbox, system_prompt
16
 
17
  load_dotenv()
18