Update app.py

app.py

@@ -38,92 +38,36 @@ def generate(
         yield output
     return output
 
-def style():
-    return """
-        <style>
-            .title {
-                font-size: 24px;
-                font-weight: bold;
-                color: #333;
-                margin-bottom: 10px;
-            }
-            .container {
-                background-color: #f8f9fa;
-                padding: 20px;
-                border-radius: 10px;
-                box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-                margin-bottom: 20px;
-            }
-            .prompt {
-                font-size: 16px;
-                margin-bottom: 10px;
-            }
-            .response {
-                font-size: 18px;
-                font-style: italic;
-                color: #666;
-            }
-            .input-container input[type="text"] {
-                width: 100%;
-                padding: 10px;
-                font-size: 16px;
-                border-radius: 5px;
-                border: 1px solid #ccc;
-            }
-            .input-container input[type="text"]:focus {
-                outline: none;
-                border-color: #4d90fe;
-            }
-        </style>
-    """
+additional_inputs=[
+    gr.Slider(
+        label="temperature",
+        value=0.9,
+        minimum=0.0,
+        maximum=1.0,
+        step=0.05,
+        interactive=True,
+        info="Higher values generate more diverse outputs",
+    ),
+    gr.Slider(
+        label="max_new_tokens",
+        value=256,
+        minimum=1,
+        maximum=2048,
+        step=1,
+        interactive=True,
+        info="Higher values allows for longer answers",
+    )
+]
+    
+mychatbot = gr.Chatbot(
+    avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
 
-def custom_interface():
-    return """
-        <div class="container">
-            <div class="title">Nova V2</div>
-            <div class="input-container">
-                <div class="prompt">User Prompt:</div>
-                <input type="text" id="user-prompt">
-            </div>
-            <div class="input-container">
-                <div class="prompt">Chat History:</div>
-                <textarea id="chat-history" rows="4"></textarea>
-            </div>
-            <div class="input-container">
-                <button onclick="handleSubmit()" style="background-color: #4CAF50; color: white; padding: 14px 20px; margin-top: 10px; border: none; border-radius: 5px; cursor: pointer;">Generate Response</button>
-            </div>
-            <div class="response" id="response"></div>
-        </div>
-        <script>
-            function handleSubmit() {
-                var userPrompt = document.getElementById('user-prompt').value;
-                var chatHistory = document.getElementById('chat-history').value;
-                var responseContainer = document.getElementById('response');
-                responseContainer.innerHTML = "Loading...";
-                var xhr = new XMLHttpRequest();
-                xhr.open('POST', '/predict');
-                xhr.setRequestHeader('Content-Type', 'application/json');
-                xhr.onload = function() {
-                    if (xhr.status === 200) {
-                        responseContainer.innerHTML = xhr.responseText;
-                    } else {
-                        responseContainer.innerHTML = 'Error: ' + xhr.statusText;
-                    }
-                };
-                xhr.send(JSON.stringify({prompt: userPrompt, history: chatHistory}));
-            }
-        </script>
-    """
+demo = gr.ChatInterface(fn=generate, 
+                        chatbot=mychatbot,
+                        title="Nova V2",
+                        additional_inputs=additional_inputs,
+                        retry_btn=None,
+                        undo_btn=None
+                       )
 
-gr.Interface(
-    generate, 
-    inputs=["text", "text", "text", "text", "text"],
-    outputs="text",
-    title="Nova V2",
-    examples=[
-        ["Hello!", "Hi there!", "What are you doing?", "Nothing much, just chatting with you.", "How are you today?"],
-    ],
-    css=style(),
-    allow_screenshot=False,
-    capture_session=True
-).launch(share=True, debug=True)
+demo.queue().launch() remove additional inputs and add a beautiful ui