Update app.py

app.py

@@ -1,136 +1,44 @@
 import gradio as gr
-from gradio import ChatMessage
-import time
 
-def slow_echo(message, history):
-    for i in range(len(message)):
-        time.sleep(0.05)
-        yield "You typed: " + message[: i + 1]
-
-demo = gr.ChatInterface(
-    slow_echo,
-    type="messages",
-    flagging_mode="manual",
-    flagging_options=["Like", "Spam", "Inappropriate", "Other"],
-    save_history=True,
-)
-
-
-# Define the main function for the app
-def main_function(purpose, help_text, export_text, extra_menu_item, chat_input, tts_input, script_input):
-    # Placeholder function to demonstrate the app
-    return {
-        "purpose_output": purpose,
-        "help_output": help_text,
-        "export_output": export_text,
-        "extra_menu_output": extra_menu_item,
-        "chat_output": chat_input,
-        "tts_output": tts_input,
-        "script_output": script_input
-    }
-
-# Define the Gradio app
-with gr.Blocks() as demo:
-    # Define input components
-    purpose_input = gr.Textbox(label="Purpose")
-    help_input = gr.Textbox(label="Help Text")
-    export_input = gr.Textbox(label="Export Text")
-    extra_menu_input = gr.Textbox(label="Extra Menu Item")
-    chat_input = gr.Textbox(label="Chat Input")
-    tts_input = gr.Textbox(label="TTS Input")
-    script_input = gr.Textbox(label="Script Input")
-
-    # Define output components
-    purpose_output = gr.Textbox(label="Purpose Output")
-    help_output = gr.Textbox(label="Help Output")
-    export_output = gr.Textbox(label="Export Output")
-    extra_menu_output = gr.Textbox(label="Extra Menu Output")
-    chat_output = gr.Textbox(label="Chat Output")
-    tts_output = gr.Textbox(label="TTS Output")
-    script_output = gr.Textbox(label="Script Output")
-
-    # Define the button to trigger the function
-    submit_button = gr.Button("Submit", variant="primary")
-
-    # Set up the event listener for the button
-    submit_button.click(
-        fn=main_function,
-        inputs=[purpose_input, help_input, export_input, extra_menu_input, chat_input, tts_input, script_input],
-        outputs=[purpose_output, help_output, export_output, extra_menu_output, chat_output, tts_output, script_output]
-    )
-
-def generate_response(history):
-    history.append(
-        ChatMessage(
-            role="user", content="What is the weather in San Francisco right now?"
-        )
-    )
-    yield history
-    time.sleep(0.25)
-    history.append(
-        ChatMessage(
-            role="assistant",
-            content="In order to find the current weather in San Francisco, I will need to use my weather tool.",
-        )
-    )
-    yield history
-    time.sleep(0.25)
-
-    history.append(
-        ChatMessage(
-            role="assistant",
-            content="API Error when connecting to weather service.",
-            metadata={"title": "💥 Error using tool 'Weather'"},
-        )
-    )
-    yield history
-    time.sleep(0.25)
-
-    history.append(
-        ChatMessage(
-            role="assistant",
-            content="I will try again",
-        )
-    )
-    yield history
-    time.sleep(0.25)
-
-    history.append(
-        ChatMessage(
-            role="assistant",
-            content="Weather 72 degrees Fahrenheit with 20% chance of rain.",
-            metadata={"title": "🛠️ Used tool 'Weather'"},
-        )
-    )
-    yield history
-    time.sleep(0.25)
-
-    history.append(
-        ChatMessage(
-            role="assistant",
-            content="Now that the API succeeded I can complete my task.",
-        )
-    )
-    yield history
-    time.sleep(0.25)
-
-    history.append(
-        ChatMessage(
-            role="assistant",
-            content="It's a sunny day in San Francisco with a current temperature of 72 degrees Fahrenheit and a 20% chance of rain. Enjoy the weather!",
-        )
-    )
-    yield history
-
-def like(evt: gr.LikeData):
-    print("User liked the response")
-    print(evt.index, evt.liked, evt.value)
+python_code = """
+def fib(n):
+    if n <= 0:
+        return 0
+    elif n == 1:
+        return 1
+    else:
+        return fib(n-1) + fib(n-2)
+"""
+
+js_code = """
+function fib(n) {
+    if (n <= 0) return 0;
+    if (n === 1) return 1;
+    return fib(n - 1) + fib(n - 2);
+}
+"""
+
+def chat(message, history):
+    if "python" in message.lower():
+        return "Type Python or JavaScript to see the code.", gr.Code(language="python", value=python_code)
+    elif "javascript" in message.lower():
+        return "Type Python or JavaScript to see the code.", gr.Code(language="javascript", value=js_code)
+    else:
+        return "Please ask about Python or JavaScript.", None
 
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(type="messages", height=500, show_copy_button=True)
-    button = gr.Button("Get San Francisco Weather")
-    button.click(generate_response, chatbot, chatbot)
-    chatbot.like(like)
-
-if __name__ == "__main__":
-    demo.launch()
+    code = gr.Code(render=False)
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("<center><h1>Write Python or JavaScript</h1></center>")
+            gr.ChatInterface(
+                chat,
+                examples=["Python", "JavaScript"],
+                additional_outputs=[code],
+                type="messages"
+            )
+        with gr.Column():
+            gr.Markdown("<center><h1>Code Artifacts</h1></center>")
+            code.render()
+
+demo.launch()