Update app.py

app.py

@@ -6,6 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+import gradio as gr
 
        
 
@@ -86,4 +87,75 @@ agent = CodeAgent(
 )
 
 
-GradioUI(agent).launch()
+# GradioUI(agent).launch()
+
+
+css = """
+.message-row {
+    justify-content: space-evenly !important;
+}
+.message-bubble-border {
+    border-radius: 6px !important;
+}
+.message-buttons-bot, .message-buttons-user {
+    right: 10px !important;
+    left: auto !important;
+    bottom: 2px !important;
+}
+.dark.message-bubble-border {
+    border-color: #15172c !important;
+}
+.dark.user {
+    background: #10132c !important;
+}
+.dark.assistant.dark, .dark.pending.dark {
+    background: #020417 !important;
+}
+"""
+
+# Create Gradio interface
+gr.ChatInterface(
+    predict,
+    title=EMOJI + " " + MODEL_NAME,
+    description=DESCRIPTION,
+    examples=[
+        ["Can you solve the equation 2x + 3 = 11 for x?"],
+        ["Write an epic poem about Ancient Rome."],
+        ["Who was the first person to walk on the Moon?"],
+        ["Use a list comprehension to create a list of squares for numbers from 1 to 10."],
+        ["Recommend some popular science fiction books."],
+        ["Can you write a short story about a time-traveling detective?"]
+    ],
+    additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False),
+    additional_inputs=[
+        gr.Textbox("Perform the task to the best of your ability.", label="System prompt"),
+        gr.Slider(0, 1, 0.8, label="Temperature"),
+        gr.Slider(128, 4096, 512, label="Max new tokens"),
+        gr.Slider(1, 80, 40, label="Top K sampling"),
+        gr.Slider(0, 2, 1.1, label="Repetition penalty"),
+        gr.Slider(0, 1, 0.95, label="Top P sampling"),
+    ],
+    theme=gr.themes.Soft(primary_hue=COLOR).set(
+        background_fill_primary_dark="#020417",
+        background_fill_secondary_dark="#020417",
+        body_background_fill_dark="#020417",
+        block_background_fill_dark="#020417",
+        block_border_width="1px",
+        block_title_background_fill_dark="#15172c",
+        input_background_fill_dark="#15172c",
+        button_secondary_background_fill_dark="#15172c",
+        border_color_accent_dark="#15172c",
+        border_color_primary_dark="#15172c",
+        color_accent_soft_dark="#10132c",
+        code_background_fill_dark="#15172c",
+    ),
+    css=css,
+    retry_btn="Retry",
+    undo_btn="Undo",
+    clear_btn="Clear",
+    submit_btn="Send",
+    chatbot=gr.Chatbot(
+        scale=1,
+        show_copy_button=True
+    )
+).queue().launch()