Update app.py

app.py

@@ -1,65 +1,68 @@
+import tkinter as tk
+from tkinter import ttk, scrolledtext, filedialog
 from transformers import pipeline
-from huggingface_hub import hf_hub_url, cached_download, HfApi
+import logging
 
-HF_TOKEN = os.environ.get("HF_TOKEN", None)
+# Logging Setup
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-def generate_code_from_model(input_text):
-    """Generates code using the bigscience/T0_3B model."""
-    model_name = 'bigscience/T0_3B'  # Choose your model
-    generator = pipeline('text-generation', model=model_name)
-    model_type = {
-        "model_type": "compositional",
-        "task": "Develop a cybersecurity tool for detecting and mitigating supply-chain attacks",
-        "history": "",
-        "history_additions": [
-            "Set the task description to 'Develop a cybersecurity tool for detecting and mitigating supply-chain attacks'",
-            "Updated the task description to 'Identify common supply-chain attack techniques'",
-            "Completed the task: Identify common supply-chain attack techniques",
-            "Updated the task description to 'Design and implement detection mechanisms for supply-chain attacks'",
-            "Completed the task: Design and implement detection mechanisms for supply-chain attacks",
-            "Updated the task description to 'Integrate threat intelligence feeds to improve attack detection'",
-            "Completed the task: Integrate threat intelligence feeds to improve attack detection",
-            "Updated the task description to 'Implement mitigation techniques to minimize the impact of supply-chain attacks'",
-            "Completed the task: Implement mitigation techniques to minimize the impact of supply-chain attacks",
-            "Updated the task description to 'Conduct testing and validation to ensure tool effectiveness'",
-            "Completed the task: Conduct testing and validation to ensure tool effectiveness",
-            "Updated the task description to 'Prepare documentation to help users understand and get started with the tool'",
-            "Completed the task: Prepare documentation to help users understand and get started with the tool",
-            "Updated the task description to 'Integrate the tool with security monitoring and incident response workflows'"
-        ],
-        "history_removals": [],
-        "model_log": [
-            "Model prompt: You are attempting to complete the task task: Develop a cybersecurity tool for detecting and mitigating supply-chain attacks Progress:  Identify common supply-chain attack techniques",
-            "Model response: Identify common supply-chain attack techniques",
-            "Model prompt: Updated the task description to 'Design and implement detection mechanisms for supply-chain attacks'",
-            "Model response: Design and implement detection mechanisms for supply-chain attacks",
-            "Model prompt: Completed the task: Design and implement detection mechanisms for supply-chain attacks Progress: Identify common supply-chain attack techniques\nDesign and implement detection mechanisms for supply-chain attacks",
-            "Model response: You have successfully completed the task: Design and implement detection mechanisms for supply-chain attacks",
-            "Model prompt: Updated the task description to 'Integrate threat intelligence feeds to improve attack detection'",
-            "Model response: Integrate threat intelligence feeds to improve attack detection",
-            "Model prompt: Completed the task: Integrate threat intelligence feeds to improve attack detection Progress: Identify common supply-chain attack techniques\nDesign and implement detection mechanisms for supply-chain attacks\nIntegrate threat intelligence feeds to improve attack detection",
-            "Model response: You have successfully completed the task: Integrate threat intelligence feeds to improve attack detection",
-            "Model prompt: Updated the task description to 'Implement mitigation techniques to minimize the impact of supply-chain attacks'",
-            "Model response: Implement mitigation techniques to minimize the impact of supply-chain attacks",
-            "Model prompt: Completed the task: Implement mitigation techniques to minimize the impact of supply-chain attacks Progress: Identify common supply-chain attack techniques\nDesign and implement detection mechanisms for supply-chain attacks\nIntegrate threat intelligence feeds to improve attack detection\nImplement mitigation techniques to minimize the impact of supply-chain attacks",
-            "Model response: You have successfully completed the task: Implement mitigation techniques to minimize the impact of supply-chain attacks",
-            "Model prompt: Updated the task description to 'Conduct testing and validation to ensure tool effectiveness'",
-            "Model response: Conduct testing and validation to ensure tool effectiveness",
-            "Model prompt: Completed the task: Conduct testing and validation to ensure tool effectiveness",
-            "Model response: Conduct testing and validation to ensure tool effectiveness",
-            "Model prompt: Updated the task description to 'Prepare documentation to help users understand and get started with the tool'",
-            "Model response: Prepare documentation to help users understand and get started with the tool",
-            "Model prompt: Completed the task: Prepare documentation to help users understand and get started with the tool Progress: Identify common supply-chain attack techniques\nDesign and implement detection mechanisms for supply-chain attacks\nIntegrate threat intelligence feeds to improve attack detection\nImplement mitigation techniques to minimize the impact of supply-chain attacks\nConduct testing and validation to ensure tool effectiveness\nPrepare documentation to help users understand and get started with the tool",
-            "Model response: You have successfully completed the task: Prepare documentation to help users understand and get started with the tool",
-            "Model prompt: Updated the task description to 'Integrate the tool with security monitoring and incident response workflows'",
-            "Model response: Integrate the tool with security monitoring and incident response workflows",
-            "Model prompt: Completed the task: Integrate the tool with security monitoring and incident response workflows Progress: Identify common supply-chain attack techniques\nDesign and implement detection mechanisms for supply-chain attacks\nIntegrate threat intelligence feeds to improve attack detection\nImplement mitigation techniques to minimize the impact of supply-chain attacks\nConduct testing and validation to ensure tool effectiveness\nPrepare documentation to help users understand and get started with the tool\nIntegrate the tool with security monitoring and incident response workflows",
-            "Model response: You have successfully completed the task: Integrate the tool with security monitoring and incident response workflows",
-            "Model prompt: Task complete. Tool development for detecting and mitigating supply-chain attacks has been completed.",
-            "Model response: Task complete. Tool development for detecting and mitigating supply-chain attacks has been completed."
-        ]
-    }
+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')
 
-    code = generator(input_text, max_length=250, num_return_sequences=1, do_sample=True)[0]['generated_text']
-    code = code.strip()
-    return code
+    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.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.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}")
+
+root = tk.Tk()
+app = CodeGeneratorApp(root)
+root.mainloop()