fix

app.py

@@ -108,51 +108,64 @@ def create_demo() -> gr.Blocks:
         
         gr.Markdown("<br>")
         
-        gr.ChatInterface(
-            fn=generate,
-            additional_inputs=[
-                gr.Textbox(
-                    label="System prompt", 
-                    lines=6,
-                    value="You are a knowledgeable philosophy professor using the Stanford Encyclopedia of Philosophy as your knowledge base. Provide clear, accurate responses using markdown formatting. Focus on philosophical concepts and maintain academic rigor while being accessible. Always cite relevant philosophers and concepts."
-                ),
-                gr.Slider(
-                    label="Max new tokens",
-                    minimum=1,
-                    maximum=MAX_MAX_NEW_TOKENS,
-                    step=1,
-                    value=DEFAULT_MAX_NEW_TOKENS,
-                ),
-                gr.Slider(
-                    label="Temperature",
-                    minimum=0.1,
-                    maximum=4.0,
-                    step=0.1,
-                    value=0.7,
-                ),
-                gr.Slider(
-                    label="Top-p (nucleus sampling)",
-                    minimum=0.05,
-                    maximum=1.0,
-                    step=0.05,
-                    value=0.9,
-                ),
-                gr.Slider(
-                    label="Top-k",
-                    minimum=1,
-                    maximum=1000,
-                    step=1,
-                    value=50,
-                ),
-                gr.Slider(
-                    label="Repetition penalty",
-                    minimum=1.0,
-                    maximum=2.0,
-                    step=0.05,
-                    value=1.1,
-                ),
-            ],
-            stop_btn=None,
+        chatbot = gr.Chatbot(
+            show_label=False,
+            avatar_images=(None, None),
+        )
+        
+        with gr.Row():
+            msg = gr.Textbox(
+                scale=4,
+                show_label=False,
+                placeholder="Enter text and press enter",
+                container=False,
+            )
+            submit = gr.Button("Submit", scale=1, variant="primary")
+        
+        system_prompt = gr.Textbox(
+            label="System prompt", 
+            lines=6,
+            value="You are a knowledgeable philosophy professor using the Stanford Encyclopedia of Philosophy as your knowledge base. Provide clear, accurate responses using markdown formatting. Focus on philosophical concepts and maintain academic rigor while being accessible. Always cite relevant philosophers and concepts."
+        )
+        
+        with gr.Accordion("Generation Parameters", open=False):
+            max_new_tokens = gr.Slider(
+                label="Max new tokens",
+                minimum=1,
+                maximum=MAX_MAX_NEW_TOKENS,
+                step=1,
+                value=DEFAULT_MAX_NEW_TOKENS,
+            )
+            temperature = gr.Slider(
+                label="Temperature",
+                minimum=0.1,
+                maximum=4.0,
+                step=0.1,
+                value=0.7,
+            )
+            top_p = gr.Slider(
+                label="Top-p (nucleus sampling)",
+                minimum=0.05,
+                maximum=1.0,
+                step=0.05,
+                value=0.9,
+            )
+            top_k = gr.Slider(
+                label="Top-k",
+                minimum=1,
+                maximum=1000,
+                step=1,
+                value=50,
+            )
+            repetition_penalty = gr.Slider(
+                label="Repetition penalty",
+                minimum=1.0,
+                maximum=2.0,
+                step=0.05,
+                value=1.1,
+            )
+
+        gr.Examples(
             examples=[
                 ["What is the trolley problem and what are its main ethical implications?"],
                 ["Can you explain Plato's Theory of Forms?"],
@@ -160,14 +173,18 @@ def create_demo() -> gr.Blocks:
                 ["How does Kant's Categorical Imperative work?"],
                 ["What is the problem of consciousness in philosophy of mind?"],
             ],
-            chatbot=gr.Chatbot(
-                show_label=False,
-                avatar_images=(None, None),
-            ),
-            autofocus=True,
-            retry_btn=None,
-            undo_btn=None,
-            clear_btn=None,
+            inputs=msg,
+        )
+
+        msg.submit(
+            generate,
+            [msg, chatbot, system_prompt, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
+            [chatbot],
+        )
+        submit.click(
+            generate,
+            [msg, chatbot, system_prompt, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
+            [chatbot],
         )
         
         gr.Markdown(LICENSE)