Update app.py

remove unnecessary elements

app.py

@@ -1,23 +1,16 @@
-import datetime
 import os
 import re
-from io import StringIO
 
 import gradio as gr
-import pandas as pd
-from huggingface_hub import upload_file
 from text_generation import Client
 
 from dialogues import DialogueTemplate
-from share_btn import (community_icon_html, loading_icon_html, share_btn_css,
-                       share_js)
 
 model2endpoint = {
     "starchat-beta": os.environ.get("API_URL", None),
 }
 model_names = list(model2endpoint.keys())
 
-
 def get_total_inputs(inputs, chatbot, preprompt, user_name, assistant_name, sep):
     past = []
     for data in chatbot:
@@ -52,19 +45,19 @@ def has_no_history(chatbot, history):
 
 
 def generate(
-    model_name,
-    system_message,
     user_message,
     chatbot,
     history,
-    temperature,
-    top_k,
-    top_p,
-    max_new_tokens,
-    repetition_penalty,
 ):
+    system_message = "Below is a conversation between a human user and a helpful AI coding assistant."
+    temperature = 0.2
+    top_k = 50
+    top_p = 0.95
+    max_new_tokens = 1024
+    repetition_penalty = 1.2
+
     client = Client(
-        model2endpoint[model_name]
+        model2endpoint["starchat-beta"]
     )
     # Don't return meaningless message when the input is empty
     if not user_message:
@@ -163,14 +156,16 @@ def process_example(args):
     return [x, y]
 
 
-title = """<h1 align="center">⭐ StarChat Saturdays 💬</h1>"""
+title = """<h1 align="center">⭐ StarChat Saturdays 💬</h1>
+<h2 align="center">Asistente de IA para estudiantes de Inteligencia Artificial</h2>
+<h3 align="center">¡Tu privacidad es nuestra prioridad! Toda la información compartida en esta conversación se elimina automáticamente una vez que salgas del chat.</h3>
+"""
 custom_css = """
 #banner-image {
     display: block;
     margin-left: auto;
     margin-right: auto;
 }
-
 #chat-message {
     font-size: 14px;
     min-height: 300px;
@@ -180,15 +175,6 @@ custom_css = """
 with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
     gr.HTML(title)
 
-    with gr.Row():
-        selected_model = gr.Radio(choices=model_names, value=model_names[0], label="Select a model")
-
-    with gr.Accordion(label="System Prompt", open=False, elem_id="parameters-accordion"):
-        system_message = gr.Textbox(
-            elem_id="system-message",
-            placeholder="Below is a conversation between a human user and a helpful AI coding assistant.",
-            show_label=False,
-        )
     with gr.Row():
         with gr.Box():
             output = gr.Markdown()
@@ -200,86 +186,16 @@ with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
             with gr.Row():
                 send_button = gr.Button("Send", elem_id="send-btn", visible=True)
 
-                # regenerate_button = gr.Button("Regenerate", elem_id="send-btn", visible=True)
-
-                clear_chat_button = gr.Button("Clear chat", elem_id="clear-btn", visible=True)
-
-            with gr.Accordion(label="Parameters", open=False, elem_id="parameters-accordion"):
-                temperature = gr.Slider(
-                    label="Temperature",
-                    value=0.2,
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.1,
-                    interactive=True,
-                    info="Higher values produce more diverse outputs",
-                )
-                top_k = gr.Slider(
-                    label="Top-k",
-                    value=50,
-                    minimum=0.0,
-                    maximum=100,
-                    step=1,
-                    interactive=True,
-                    info="Sample from a shortlist of top-k tokens",
-                )
-                top_p = gr.Slider(
-                    label="Top-p (nucleus sampling)",
-                    value=0.95,
-                    minimum=0.0,
-                    maximum=1,
-                    step=0.05,
-                    interactive=True,
-                    info="Higher values sample more low-probability tokens",
-                )
-                max_new_tokens = gr.Slider(
-                    label="Max new tokens",
-                    value=1024,
-                    minimum=0,
-                    maximum=2048,
-                    step=4,
-                    interactive=True,
-                    info="The maximum numbers of new tokens",
-                )
-                repetition_penalty = gr.Slider(
-                    label="Repetition Penalty",
-                    value=1.2,
-                    minimum=0.0,
-                    maximum=10,
-                    step=0.1,
-                    interactive=True,
-                    info="The parameter for repetition penalty. 1.0 means no penalty.",
-                )
-            # with gr.Group(elem_id="share-btn-container"):
-            #     community_icon = gr.HTML(community_icon_html, visible=True)
-            #     loading_icon = gr.HTML(loading_icon_html, visible=True)
-            # share_button = gr.Button("Share to community", elem_id="share-btn", visible=True)
-            with gr.Row():
-                gr.Examples(
-                    examples=examples,
-                    inputs=[user_message],
-                    cache_examples=False,
-                    fn=process_example,
-                    outputs=[output],
-                )
-
     history = gr.State([])
-    # To clear out "message" input textbox and use this to regenerate message
+
     last_user_message = gr.State("")
 
     user_message.submit(
         generate,
         inputs=[
-            selected_model,
-            system_message,
             user_message,
             chatbot,
             history,
-            temperature,
-            top_k,
-            top_p,
-            max_new_tokens,
-            repetition_penalty,
         ],
         outputs=[chatbot, history, last_user_message, user_message],
     )
@@ -287,22 +203,11 @@ with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
     send_button.click(
         generate,
         inputs=[
-            selected_model,
-            system_message,
             user_message,
             chatbot,
             history,
-            temperature,
-            top_k,
-            top_p,
-            max_new_tokens,
-            repetition_penalty,
         ],
         outputs=[chatbot, history, last_user_message, user_message],
     )
 
-    clear_chat_button.click(clear_chat, outputs=[chatbot, history])
-    selected_model.change(clear_chat, outputs=[chatbot, history])
-    # share_button.click(None, [], [], _js=share_js)
-
-demo.queue(concurrency_count=16).launch()
+demo.queue(concurrency_count=16).launch()