UI_Runner.py

import tkinter as tk
from tkinter import ttk, messagebox, filedialog, scrolledtext
from requirements import example_system_requirements
from system_designs import example_system_designs
from verification_requirements import example_verification_requirements
import api_integration

class SystemModelApp:
    def __init__(self, root: tk.Tk):
        self.root = root
        self.root.title("System Model Analysis Interface")
        self.root.geometry("800x600")

        # Variables
        self.api_key = tk.StringVar()
        self.system_requirements = tk.StringVar()
        self.results = ""
        self.verification_help = tk.BooleanVar(value=False)
        self.traceability_help = tk.BooleanVar(value=False)
        self.verification_conditions = tk.BooleanVar(value=False)

        self.create_layout()

    def create_layout(self):
        # Create main container with padding
        main_container = ttk.Frame(self.root, padding="10")
        main_container.pack(fill='both', expand=True)

        # API Configuration Section
        api_section = ttk.LabelFrame(main_container, text="API Configuration", padding="5")
        api_section.pack(fill='x', pady=(0, 10))

        ttk.Label(api_section, text="Gemini API Key:").pack(pady=2)
        ttk.Entry(api_section, textvariable=self.api_key, width=50).pack(pady=2)
        ttk.Button(api_section, text="Initialize API", command=self.initialize_api).pack(pady=2)

        # System Requirements Section
        requirements_section = ttk.LabelFrame(main_container, text="System Requirements", padding="5")
        requirements_section.pack(fill='x', pady=(0, 10))

        ttk.Label(requirements_section, text="Enter System Requirements:").pack(pady=2)
        self.requirements_text = tk.Text(requirements_section, height=10, width=60)
        self.requirements_text.pack(pady=2)

        ttk.Button(requirements_section, text="Submit Requirements", command=self.submit_requirements).pack(pady=2)

        # Verification Help Section
        verification_section = ttk.LabelFrame(main_container, text="Verification Help", padding="5")
        verification_section.pack(fill='x', pady=(0, 10))

        ttk.Label(verification_section, text="Would you like help with verification (Y/N)?").pack(pady=2)
        ttk.Checkbutton(verification_section, variable=self.verification_help).pack(pady=2)

        # Traceability Help Section
        traceability_section = ttk.LabelFrame(main_container, text="Traceability Help", padding="5")
        traceability_section.pack(fill='x', pady=(0, 10))

        ttk.Label(traceability_section, text="Would you like traceability and proof (Y/N)?").pack(pady=2)
        ttk.Checkbutton(traceability_section, variable=self.traceability_help).pack(pady=2)

        # Verification Conditions Section
        conditions_section = ttk.LabelFrame(main_container, text="Verification Conditions", padding="5")
        conditions_section.pack(fill='x', pady=(0, 10))

        ttk.Label(conditions_section, text="Would you like verification conditions (Y/N)?").pack(pady=2)
        ttk.Checkbutton(conditions_section, variable=self.verification_conditions).pack(pady=2)

        # Run Analysis Button
        ttk.Button(main_container, text="Run Analysis", command=self.run_analysis).pack(pady=10)

        # Check Out Results Button
        ttk.Button(main_container, text="Check Out Results", command=self.show_results).pack(pady=5)

        # Download Results Button
        ttk.Button(main_container, text="Download Results", command=self.download_results).pack(pady=5)

    def initialize_api(self):
        if api_integration.initialize_api(self.api_key.get()):
            messagebox.showinfo("Success", "API initialized successfully")
        else:
            messagebox.showerror("Error", "Failed to initialize API")

    def submit_requirements(self):
        self.system_requirements = self.requirements_text.get("1.0", tk.END)
        messagebox.showinfo("Info", "System requirements submitted successfully")

    def run_analysis(self):
        try:
            result = api_integration.generate_system_designs(self.system_requirements, example_system_requirements, example_system_designs)
            self.results = result

            if self.verification_help.get():
                verification_result = api_integration.create_verification_requirements_models(self.system_requirements, example_system_requirements, example_verification_requirements, example_system_designs)
                self.results += "\n\n" + verification_result

            if self.traceability_help.get():
                traceability_result = api_integration.get_traceability(self.system_requirements, example_system_requirements, example_system_designs)
                self.results += "\n\n" + traceability_result

            if self.verification_conditions.get():
                conditions_result = api_integration.get_verification_conditions(self.system_requirements, example_system_requirements, example_verification_requirements, example_system_designs)
                self.results += "\n\n" + conditions_result

            messagebox.showinfo("Success", "Analysis completed successfully")
        except Exception as e:
            messagebox.showerror("Error", f"Analysis failed: {str(e)}")

    def show_results(self):
        # Create a new window to display the results
        results_window = tk.Toplevel(self.root)
        results_window.title("Analysis Results")
        results_window.geometry("800x600")

        # Create a scrolled text widget
        results_text = scrolledtext.ScrolledText(results_window, width=100, height=30, wrap=tk.WORD)
        results_text.pack(pady=10, fill='both', expand=True)
        results_text.insert(tk.END, self.results)
        results_text.config(state='disabled')

    def download_results(self):
        # Open a file dialog to save the results
        file_path = filedialog.asksaveasfilename(defaultextension=".txt", filetypes=[("Text files", "*.txt")])
        if file_path:
            with open(file_path, "w") as file:
                file.write(self.results)
            messagebox.showinfo("Success", "Results downloaded successfully")

if __name__ == "__main__":
    root = tk.Tk()
    app = SystemModelApp(root)
    root.mainloop()