Update app.py

app.py

@@ -1,13 +1,20 @@
-import gradio as gr
+import streamlit as st
 import os
 import subprocess
 import time
 
-def chat_with_code(history, user_input):
+st.set_page_config(page_title="Code Interpreter & Terminal Chat", page_icon="🤖")
+
+st.title("Code Interpreter & Terminal Chat")
+st.markdown("Ask questions, write code, and run terminal commands!")
+
+chat_history = []
+
+def chat_with_code(user_input):
   """
   Handles user input and processes it through code interpreter and terminal.
   """
-  history.append((user_input, None))  # Add user input to history
+  chat_history.append((user_input, None))  # Add user input to history
 
   try:
     # Attempt to execute code
@@ -18,11 +25,11 @@ def chat_with_code(history, user_input):
       # Attempt to execute terminal command
       output = execute_terminal(user_input)
 
-    history[-1] = (user_input, output)  # Update history with output
+    chat_history[-1] = (user_input, output)  # Update history with output
   except Exception as e:
-    history[-1] = (user_input, f"Error: {e}")
+    chat_history[-1] = (user_input, f"Error: {e}")
 
-  return history
+  return chat_history
 
 def execute_code(code):
   """
@@ -46,9 +53,24 @@ def execute_terminal(command):
     output += f"\nError: {stderr.decode('utf-8').strip()}"
   return output
 
-# Create Gradio interface
-iface = gr.ChatInterface(chat_with_code, 
-                        title="Code Interpreter & Terminal Chat",
-                        description="Ask questions, write code, and run terminal commands!")
+# Display chat history
+for message in chat_history:
+  with st.chat_message(message[0]):
+    if message[1]:
+      st.write(message[1])
+
+# User input area
+user_input = st.text_input("Enter your message or code:", key="input")
+
+# Process user input when Enter is pressed
+if user_input:
+  chat_history = chat_with_code(user_input)
+
+  # Display updated chat history
+  for message in chat_history:
+    with st.chat_message(message[0]):
+      if message[1]:
+        st.write(message[1])
 
-iface.launch(server_port=7861)
+  # Clear the input field
+  st.session_state.input = ""