prithivMLmods commited on
Commit
48b9513
1 Parent(s): c72a7a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -34
app.py CHANGED
@@ -3,7 +3,7 @@ from openai import OpenAI
3
  import os
4
 
5
  css = '''
6
- .gradio-container{max-width: 670px !important}
7
  h1{text-align:center}
8
  footer {
9
  visibility: hidden
@@ -71,7 +71,7 @@ mood_prompts = {
71
 
72
  def respond(
73
  message,
74
- history: list[tuple[str, str]],
75
  system_message,
76
  max_tokens,
77
  temperature,
@@ -107,39 +107,42 @@ def respond(
107
  response += token
108
  yield response
109
 
110
- # Layout with mood dropdown in a separate row
111
- with gr.Blocks(css=css, theme="allenai/gradio-theme") as demo:
112
- gr.Markdown("# Chat Interface")
113
-
114
- with gr.Row():
115
- with gr.Column():
116
- message_input = gr.Textbox(placeholder="Type your message here...", label="Message")
117
- history_input = gr.Textbox(placeholder="Chat history (if any)...", label="History")
118
- system_message_input = gr.Textbox(value="", label="System message")
119
- max_tokens_slider = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
120
- temperature_slider = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
121
- top_p_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
 
122
 
123
- with gr.Row():
124
- mood_dropdown = gr.Dropdown(
125
- choices=list(mood_prompts.keys()),
126
- label="Mood",
127
- value="Casual"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
128
  )
129
-
130
- chat_interface = gr.ChatInterface(
131
- fn=respond,
132
- inputs=[
133
- message_input,
134
- history_input,
135
- system_message_input,
136
- max_tokens_slider,
137
- temperature_slider,
138
- top_p_slider,
139
- mood_dropdown
140
- ],
141
- outputs=gr.Textbox(label="Response")
142
- )
143
 
144
  if __name__ == "__main__":
145
- demo.launch()
 
3
  import os
4
 
5
  css = '''
6
+ .gradio-container{max-width: 1000px !important}
7
  h1{text-align:center}
8
  footer {
9
  visibility: hidden
 
71
 
72
  def respond(
73
  message,
74
+ history,
75
  system_message,
76
  max_tokens,
77
  temperature,
 
107
  response += token
108
  yield response
109
 
110
+ # Define the layout
111
+ def create_interface():
112
+ with gr.Blocks(css=css, theme="allenai/gradio-theme") as demo:
113
+ gr.Markdown("# Chat Interface")
114
+
115
+ with gr.Row():
116
+ with gr.Column():
117
+ message_input = gr.Textbox(placeholder="Type your message here...", label="Message")
118
+ history_input = gr.Textbox(placeholder="Chat history (if any)...", label="History")
119
+ system_message_input = gr.Textbox(value="", label="System message")
120
+ max_tokens_slider = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
121
+ temperature_slider = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
122
+ top_p_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
123
 
124
+ with gr.Row():
125
+ mood_dropdown = gr.Dropdown(
126
+ choices=list(mood_prompts.keys()),
127
+ label="Mood",
128
+ value="Casual"
129
+ )
130
+
131
+ chat_interface = gr.Interface(
132
+ fn=respond,
133
+ inputs=[
134
+ message_input,
135
+ history_input,
136
+ system_message_input,
137
+ max_tokens_slider,
138
+ temperature_slider,
139
+ top_p_slider,
140
+ mood_dropdown
141
+ ],
142
+ outputs=gr.Textbox(label="Response")
143
  )
144
+
145
+ chat_interface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
146
 
147
  if __name__ == "__main__":
148
+ create_interface()