Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+import torch
 import subprocess
 
 # Load a code-generation model with support for multiple languages
@@ -23,18 +24,24 @@ def execute_code(code, language):
             return str(e)
     return "Code execution only supported for Python."
 
+def generate_and_execute(description, language):
+    code = generate_code(description, language)
+    output = execute_code(code, language) if language == "Python" else "Execution not supported for this language."
+    return code, output
+
 # Create a Gradio interface with language selection and execution
 iface = gr.Interface(
-    fn=generate_code,
+    fn=generate_and_execute,
     inputs=[
         gr.Textbox(lines=5, placeholder="Describe your coding task..."),
         gr.Dropdown(choices=["Python", "JavaScript", "Java"], label="Programming Language")
     ],
-    outputs=[gr.Code(), gr.Textbox(label="Execution Output")],
+    outputs=[gr.Code(label="Generated Code"), gr.Textbox(label="Execution Output")],
     title="Multi-Language Text-to-Code AI",
     description="Convert natural language descriptions into code in different programming languages! Run Python code directly in the app.",
     theme="default",
-    allow_flagging="never"
+    allow_flagging="never",
+    live=True
 )
 
 # Launch the app