cosmosai471 commited on
Commit
98082d2
·
verified ·
1 Parent(s): 67b8b0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -2,9 +2,9 @@ import gradio as gr
2
  from lunacode import smart_luna_answer, text_to_speech_luna
3
 
4
  # TEXT CHAT
5
- def text_chat(user_input, max_tokens, system_message, history):
6
  response = smart_luna_answer(user_input, max_tokens=max_tokens)
7
- history.append((user_input, response))
8
  return history, history
9
 
10
  # VOICE CHAT
@@ -23,37 +23,37 @@ def voice_chat(audio, max_tokens):
23
  audio_path = text_to_speech_luna(response_text)
24
  return response_text, audio_path
25
 
26
- # INTERFACE 1: CHAT
27
  chat_interface = gr.Interface(
28
  fn=text_chat,
29
  inputs=[
30
- gr.Textbox(label="Your Message"),
31
- gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max new tokens"),
32
- gr.Textbox(value="You are a helpful AI assistant.", label="System message"),
33
- gr.State([]) # chat history
34
  ],
35
  outputs=[
36
  gr.Chatbot(label="Luna Chat"),
37
  gr.State([])
38
  ],
39
- title="Luna Chat Assistant"
40
  )
41
 
42
- # INTERFACE 2: VOICE
43
  voice_interface = gr.Interface(
44
  fn=voice_chat,
45
  inputs=[
46
  gr.Audio( type="filepath", label="Speak to Luna"),
47
- gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max new tokens")
48
  ],
49
  outputs=[
50
  gr.Textbox(label="Luna's Response"),
51
  gr.Audio(label="Luna Speaks")
52
  ],
53
- title="Luna Voice Assistant"
54
  )
55
 
56
- # TABS
57
  demo = gr.TabbedInterface(
58
  [chat_interface, voice_interface],
59
  ["Chat with Luna", "Voice with Luna"]
 
2
  from lunacode import smart_luna_answer, text_to_speech_luna
3
 
4
  # TEXT CHAT
5
+ def text_chat(user_input, history, max_tokens, system_message):
6
  response = smart_luna_answer(user_input, max_tokens=max_tokens)
7
+ history = history + [[user_input, response]]
8
  return history, history
9
 
10
  # VOICE CHAT
 
23
  audio_path = text_to_speech_luna(response_text)
24
  return response_text, audio_path
25
 
26
+ # CHAT INTERFACE
27
  chat_interface = gr.Interface(
28
  fn=text_chat,
29
  inputs=[
30
+ gr.Textbox(label="Your Message", placeholder="Ask me anything..."),
31
+ gr.State([]), # Chat history
32
+ gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max tokens"),
33
+ gr.Textbox(value="You are a helpful AI assistant.", label="System Prompt"),
34
  ],
35
  outputs=[
36
  gr.Chatbot(label="Luna Chat"),
37
  gr.State([])
38
  ],
39
+ title="Luna Chat"
40
  )
41
 
42
+ # VOICE INTERFACE
43
  voice_interface = gr.Interface(
44
  fn=voice_chat,
45
  inputs=[
46
  gr.Audio( type="filepath", label="Speak to Luna"),
47
+ gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max tokens")
48
  ],
49
  outputs=[
50
  gr.Textbox(label="Luna's Response"),
51
  gr.Audio(label="Luna Speaks")
52
  ],
53
+ title="Luna Voice"
54
  )
55
 
56
+ # COMBINE INTO TABS
57
  demo = gr.TabbedInterface(
58
  [chat_interface, voice_interface],
59
  ["Chat with Luna", "Voice with Luna"]