aaaaa

app.py

@@ -123,29 +123,45 @@ custom_css = """
 }
 """
 
-# Define the interface
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("<h1 style='text-align: center;'>🌟 Fancy AI Chatbot 🌟</h1>")
-    gr.Markdown("Interact with the AI chatbot using customizable settings below.")
-
-    with gr.Row():
-        system_message = gr.Textbox(value="You are a friendly Chatbot.", label="System message", interactive=True)
-        use_local_model = gr.Checkbox(label="Use Local Model", value=False)
-
-    with gr.Row():
-        max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-        temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
-        top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
-
-    chat_history = gr.Chatbot(label="Chat")
-
-    user_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
-
-    cancel_button = gr.Button("Cancel Inference", variant="danger")
-
-    # Adjusted to ensure history is maintained and passed correctly
-    user_input.submit(respond, [user_input, chat_history, system_message, max_tokens, temperature, top_p, use_local_model], chat_history)
-    cancel_button.click(cancel_inference)
+# # Define the interface
+# with gr.Blocks(css=custom_css) as demo:
+#     gr.Markdown("<h1 style='text-align: center;'>🌟 Fancy AI Chatbot 🌟</h1>")
+#     gr.Markdown("Interact with the AI chatbot using customizable settings below.")
+
+#     with gr.Row():
+#         system_message = gr.Textbox(value="You are a friendly Chatbot.", label="System message", interactive=True)
+#         use_local_model = gr.Checkbox(label="Use Local Model", value=False)
+
+#     with gr.Row():
+#         max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
+#         temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
+#         top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
+
+#     chat_history = gr.Chatbot(label="Chat")
+
+#     user_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
+
+#     cancel_button = gr.Button("Cancel Inference", variant="danger")
+
+#     # Adjusted to ensure history is maintained and passed correctly
+#     user_input.submit(respond, [user_input, chat_history, system_message, max_tokens, temperature, top_p, use_local_model], chat_history)
+#     cancel_button.click(cancel_inference)
+
+# if __name__ == "__main__":
+#     demo.launch(share=False)  # Remove share=True because it's not supported on HF Spaces
+
+demo = gr.ChatInterface(
+    fn=respond,
+    additional_inputs=[
+        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
+        gr.Checkbox(label="Use Local Model", value=False)
+    ],
+    cancel=gr.Button("Cancel Inference", variant="danger"),
+    css=custom_css  # Apply the custom CSS styling
+)
 
 if __name__ == "__main__":
-    demo.launch(share=False)  # Remove share=True because it's not supported on HF Spaces
+    demo.launch(share=True)