Update app.py

app.py

@@ -1,71 +1,46 @@
-import tkinter as tk
-import tkinter.ttk as ttk
-from tkinter.scrolledtext import ScrolledText
-from tkinter import filedialog
+import streamlit as st
 from transformers import pipeline
 import logging
 
 # Logging Setup
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-class CodeGeneratorApp:
-    def __init__(self, master):
-        self.master = master
-        master.title("Advanced Code Generator")
-        self.create_widgets()
-        self.generator = pipeline('text-generation', model='bigscience/T0_3B')
-
-    def create_widgets(self):
-        # Input Frame
-        input_frame = ttk.LabelFrame(self.master, text="Task Description")
-        input_frame.pack(padx=10, pady=10)
-        self.input_area = ScrolledText(input_frame, wrap=tk.WORD, width=80, height=10)
-        self.input_area.pack(padx=5, pady=5)
-
-        # Options Frame
-        options_frame = ttk.LabelFrame(self.master, text="Options")
-        options_frame.pack(padx=10, pady=5)
-        ttk.Label(options_frame, text="Max Length:").grid(row=0, column=0, sticky="w")
-        self.max_length_var = tk.IntVar(value=250)
-        ttk.Entry(options_frame, textvariable=self.max_length_var, width=5).grid(row=0, column=1)
-        self.temp_var = tk.DoubleVar(value=0.7)
-        ttk.Label(options_frame, text="Temperature:").grid(row=1, column=0, sticky="w")
-        ttk.Entry(options_frame, textvariable=self.temp_var, width=5).grid(row=1, column=1)
-
-        # Buttons
-        ttk.Button(self.master, text="Generate Code", command=self.generate_code).pack(pady=10)
-        ttk.Button(self.master, text="Save Code", command=self.save_code).pack()
-
-        # Output Frame
-        output_frame = ttk.LabelFrame(self.master, text="Generated Code")
-        output_frame.pack(padx=10, pady=10, expand=True, fill="both")
-        self.output_area = ScrolledText(output_frame, wrap=tk.WORD, width=80)
-        self.output_area.pack(padx=5, pady=5, expand=True, fill="both")
-        
-    def generate_code(self):
-        input_text = self.input_area.get("1.0", tk.END).strip()
-        max_length = self.max_length_var.get()
-        temperature = self.temp_var.get()
-
-        try:
-            logging.info(f"Generating code with input: {input_text}")  # Log the input
-            prompt = f"Develop code for the following task: {input_text}"
-            code = self.generator(prompt, max_length=max_length, num_return_sequences=1, temperature=temperature)[0]['generated_text'].strip()
-            self.output_area.delete("1.0", tk.END)
-            self.output_area.insert(tk.END, code)
-            logging.info("Code generation completed successfully.")  # Log successful generation
-        except Exception as e:
-            logging.error(f"Error generating code: {e}")
-            self.output_area.insert(tk.END, f"Error generating code: {e}")
-
-    def save_code(self):
-        file_path = filedialog.asksaveasfilename(defaultextension=".py")
-        if file_path:
-            with open(file_path, "w") as file:
-                file.write(self.output_area.get("1.0", tk.END))
-            logging.info(f"Code saved to {file_path}")
-
-if __name__ == "__main__":
-    root = tk.Tk()
-    app = CodeGeneratorApp(root)
-    root.mainloop()
+# Initialize the generator
+generator = pipeline('text-generation', model='bigscience/T0_3B')
+
+def generate_code(task_description, max_length, temperature):
+    try:
+        logging.info(f"Generating code with input: {task_description}")
+        prompt = f"Develop code for the following task: {task_description}"
+        code = generator(prompt, max_length=max_length, num_return_sequences=1, temperature=temperature)[0]['generated_text'].strip()
+        logging.info("Code generation completed successfully.")
+        return code
+    except Exception as e:
+        logging.error(f"Error generating code: {e}")
+        return f"Error generating code: {e}"
+
+st.title("Advanced Code Generator")
+
+# Input Section
+st.header("Task Description")
+task_description = st.text_area("Describe the task for which you need code:", height=150)
+
+# Options Section
+st.header("Options")
+max_length = st.slider("Max Length", min_value=50, max_value=1000, value=250, step=50)
+temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.7, step=0.1)
+
+# Generate Code Button
+if st.button("Generate Code"):
+    if task_description:
+        generated_code = generate_code(task_description, max_length, temperature)
+        st.header("Generated Code")
+        st.code(generated_code, language='python')
+
+# Save Code Section (Optional)
+if 'generated_code' in locals():
+    if st.button("Save Code"):
+        file_name = st.text_input("Enter file name to save:", value="generated_code.py")
+        with open(file_name, "w") as file:
+            file.write(generated_code)
+        st.success(f"Code saved to {file_name}")