Update app.py

app.py

@@ -37,6 +37,7 @@ class CodeExecutor:
         self.execution_complete = False
 
     def custom_input(self, prompt=''):
+        self.output += prompt
         self.waiting_for_input = True
         while self.waiting_for_input:
             pass  # Wait until input is provided
@@ -68,6 +69,10 @@ class CodeExecutor:
         self.waiting_for_input = False
 
     def get_output(self):
+        if sys.stdout and hasattr(sys.stdout, "getvalue"):
+            self.output += sys.stdout.getvalue()
+            sys.stdout.seek(0)
+            sys.stdout.truncate(0)
         return self.output
 
     def is_execution_complete(self):
@@ -79,35 +84,38 @@ def start_execution(code, session_id):
     executor = CodeExecutor()
     executors[session_id] = executor
     executor.execute_code(code)
-    return "Execution started. Waiting for output..."
+    return "Code is running...", gr.update(visible=True), gr.update(visible=True)
 
 def provide_input(user_input, session_id):
     executor = executors.get(session_id)
     if executor and not executor.is_execution_complete():
         executor.provide_input(user_input)
-        return "Input received."
+        return ""
     else:
         return "No execution in progress or execution has completed."
 
-def get_output(session_id):
+def poll_output(session_id):
     executor = executors.get(session_id)
     if executor:
         output = executor.get_output()
         if executor.is_execution_complete():
             del executors[session_id]
-        return output
+            return output, gr.update(visible=False), gr.update(visible=False)
+        else:
+            return output, gr.update(visible=True), gr.update(visible=True)
     else:
-        return "No execution in progress."
+        return "No execution in progress.", gr.update(visible=False), gr.update(visible=False)
 
 def chat(user_input, history):
     response = llm_inference(user_input)
-    history.append((user_input, response))
+    history.append({"role": "user", "content": user_input})
+    history.append({"role": "assistant", "content": response})
     return history, history
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🐍 Python Helper Chatbot")
     with gr.Tab("Chat"):
-        chatbot = gr.Chatbot()
+        chatbot = gr.Chatbot(type='messages')
         msg = gr.Textbox(placeholder="Type your message here...")
         msg.submit(chat, inputs=[msg, chatbot], outputs=[chatbot, chatbot])
     with gr.Tab("Interpreter"):
@@ -120,30 +128,27 @@ with gr.Blocks() as demo:
         session_id = gr.State()
 
         def run_code(code):
-            from uuid import uuid4
-            session = str(uuid4())
-            session_id.set(session)
-            start_execution(code, session)
-            return "Code is running...", gr.update(visible=True), gr.update(visible=True)
+            import uuid
+            session = str(uuid.uuid4())
+            session_id.value = session
+            return_values = start_execution(code, session)
+            return return_values
 
         def send_input(user_input):
-            session = session_id.get()
+            session = session_id.value
             provide_input(user_input, session)
             return ""
 
         def update_output():
-            session = session_id.get()
-            if session:
-                output = get_output(session)
-                return output
-            else:
-                return "No execution in progress."
+            session = session_id.value
+            output, input_visible, button_visible = poll_output(session)
+            return output, input_visible, button_visible
 
         run_button.click(run_code, inputs=code_input, outputs=[output_box, input_box, send_input_button])
         send_input_button.click(send_input, inputs=input_box, outputs=input_box)
-        output_box.change(update_output, outputs=output_box).every(1)
+        demo.load(update_output, outputs=[output_box, input_box, send_input_button], every=1)
     with gr.Tab("Logs"):
         gr.Markdown("### 📜 Logs")
         log_output = gr.Textbox(label="Logs", lines=10, interactive=False)
 
-demo.launch()
+demo.queue().launch()