Update app.py

app.py

@@ -4,7 +4,7 @@ import time
 import random
 
 # Load the model and tokenizer
-model_id = "microsoft/phi-2"  # Change to your desired model
+model_id = "microsoft/phi-2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 
@@ -103,49 +103,70 @@ h1, h2 {
 }
 """
 
-iface = gr.Interface(
-    fn=get_response,
-    inputs=[
-        gr.Textbox(label="Your message:", lines=5, placeholder="Ask me anything...", show_label=True),
-        gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, step=0.1, value=0.7),
-        gr.Slider(label="Top p", minimum=0.1, maximum=1.0, step=0.1, value=0.9),
-        gr.Slider(label="Top k", minimum=1, maximum=100, step=1, value=50),
-        gr.Slider(label="Max length", minimum=10, maximum=1000, step=10, value=250),
-    ],
-    outputs=[
-        gr.TextArea(label="AI Response:", lines=10),
-        gr.HTML(elem_id="analysis"),
-    ],
-    title="NVIDIA AI Chat",
-    description="Engage in a conversation with our advanced AI model. Customize the response using various parameters.",
-    theme="default",
-    css=css,
-    layout="vertical",
-    allow_flagging="never",
-)
-
-# --- Dynamic Background ---
-
-def update_background():
-    while True:
-        r = random.randint(0, 255)
-        g = 255  # Keep the green component constant
-        b = random.randint(0, 255)
-        iface.root.style.background_color = f"rgb({r}, {g}, {b})"
-        time.sleep(1)
-
-# Start a separate thread to update the background color
-gr.Interface.update(update_background, inputs=[], outputs=[], live=True)
-
-# --- Analysis Logic ---
-
-def update_analysis(response):
-    analysis = analyze_text(response)
-    analysis_str = f"<div class='analysis-container'>Number of characters: {analysis['Number of characters']}<br>" \
-                  f"Number of words: {analysis['Number of words']}<br>" \
-                  f"Number of tokens: {analysis['Number of tokens']}</div>"
-    iface.update(analysis=analysis_str, live=True)
-
-iface.outputs[0].postprocess = update_analysis
+with gr.Blocks() as iface:
+    gr.Markdown(
+        """
+        # NVIDIA AI Chat
+        Engage in a conversation with our advanced AI model. Customize the response using various parameters.
+        """
+    )
+
+    with gr.Row():
+        input_text = gr.Textbox(
+            label="Your message:", lines=5, placeholder="Ask me anything...", show_label=True
+        )
+        gr.HTML(
+            f"""
+            <style>
+                .gradio-textbox {
+                    background-color: #111111;
+                    color: #00FF00;
+                    border: 1px solid #00FF00;
+                }
+            </style>
+            """, elem_id="textbox_styles"
+        )
+
+    with gr.Row():
+        temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, step=0.1, value=0.7)
+        top_p = gr.Slider(label="Top p", minimum=0.1, maximum=1.0, step=0.1, value=0.9)
+
+    with gr.Row():
+        top_k = gr.Slider(label="Top k", minimum=1, maximum=100, step=1, value=50)
+        max_length = gr.Slider(label="Max length", minimum=10, maximum=1000, step=10, value=250)
+
+    with gr.Row():
+        submit_button = gr.Button(value="Submit")
+
+    with gr.Row():
+        response = gr.TextArea(label="AI Response:", lines=10)
+        analysis_html = gr.HTML(elem_id="analysis")
+
+    submit_button.click(fn=get_response, inputs=[input_text, temperature, top_p, top_k, max_length], outputs=[response])
+    response.change(fn=update_analysis, inputs=[response], outputs=[analysis_html])
+
+    # --- Dynamic Background ---
+
+    def update_background():
+        while True:
+            r = random.randint(0, 255)
+            g = 255  # Keep the green component constant
+            b = random.randint(0, 255)
+            iface.root.style.background_color = f"rgb({r}, {g}, {b})"
+            time.sleep(1)
+
+    # Start a separate thread to update the background color
+    gr.Interface.update(update_background, inputs=[], outputs=[], live=True)
+
+    # --- Analysis Logic ---
+
+    def update_analysis(response):
+        analysis = analyze_text(response)
+        analysis_str = f"<div class='analysis-container'>Number of characters: {analysis['Number of characters']}<br>" \
+                      f"Number of words: {analysis['Number of words']}<br>" \
+                      f"Number of tokens: {analysis['Number of tokens']}</div>"
+        iface.update(analysis=analysis_str, live=True)
+
+    iface.outputs[0].postprocess = update_analysis
 
 iface.launch(debug=True)