late night working
This commit is contained in:
parent
d44deec338
commit
e9315cb55b
3 changed files with 139 additions and 80 deletions
Binary file not shown.
|
@ -3,9 +3,10 @@ from tkinter import ttk, filedialog, messagebox
|
|||
from PIL import Image, ImageTk
|
||||
import json
|
||||
import os
|
||||
from datetime import datetime
|
||||
from datetime import datetime, date
|
||||
from plant_model import PlantGrowthModel
|
||||
from data_handler import DataHandler
|
||||
from tkcalendar import DateEntry
|
||||
|
||||
class PlantGrowthDashboard:
|
||||
def __init__(self, root):
|
||||
|
@ -35,10 +36,10 @@ class PlantGrowthDashboard:
|
|||
'humidity': 65.0,
|
||||
'soil_acidity': 6.5,
|
||||
'pressure': 1013.25,
|
||||
'brightness': 50000.0,
|
||||
'brightness': 30,
|
||||
'nutrients': 75.0,
|
||||
'water': 80.0,
|
||||
'co2': 400.0
|
||||
'co2': 850
|
||||
}
|
||||
|
||||
self.env_params = {
|
||||
|
@ -53,7 +54,6 @@ class PlantGrowthDashboard:
|
|||
}
|
||||
|
||||
self.setup_ui()
|
||||
self.update_prediction()
|
||||
|
||||
def setup_ui(self):
|
||||
# Main container with square layout
|
||||
|
@ -94,8 +94,6 @@ class PlantGrowthDashboard:
|
|||
|
||||
ttk.Radiobutton(mode_frame, text="Controlled", variable=self.ambient_mode,
|
||||
value="controlled", command=self.on_mode_change).pack(anchor=tk.W)
|
||||
ttk.Radiobutton(mode_frame, text="Semi-Controlled", variable=self.ambient_mode,
|
||||
value="semi-controlled", command=self.on_mode_change).pack(anchor=tk.W)
|
||||
ttk.Radiobutton(mode_frame, text="Open", variable=self.ambient_mode,
|
||||
value="open", command=self.on_mode_change).pack(anchor=tk.W)
|
||||
|
||||
|
@ -109,12 +107,12 @@ class PlantGrowthDashboard:
|
|||
param_labels = {
|
||||
'temperature': '🌡️ Temp (°C)',
|
||||
'humidity': '💧 Humidity (%)',
|
||||
'soil_acidity': '🧪 pH',
|
||||
'pressure': '🌬️ Pressure',
|
||||
'brightness': '☀️ Light',
|
||||
'soil_acidity': '🧪 (pH)',
|
||||
'pressure': '🌬️ Pressure (Pa)',
|
||||
'brightness': '☀️ Light (DLI)',
|
||||
'nutrients': '🌿 Nutrients (%)',
|
||||
'water': '💦 Water (%)',
|
||||
'co2': '🫧 CO2'
|
||||
'co2': '🫧 CO2 (ppm)'
|
||||
}
|
||||
|
||||
# Define bounds for each parameter (min, max)
|
||||
|
@ -123,10 +121,10 @@ class PlantGrowthDashboard:
|
|||
'humidity': (20, 90),
|
||||
'soil_acidity': (4.0, 9.0),
|
||||
'pressure': (950, 1100),
|
||||
'brightness': (100, 100000),
|
||||
'brightness': (5, 50),
|
||||
'nutrients': (0, 100),
|
||||
'water': (10, 100),
|
||||
'co2': (300, 1000)
|
||||
'co2': (400, 1200)
|
||||
}
|
||||
|
||||
row = 4
|
||||
|
@ -158,6 +156,32 @@ class PlantGrowthDashboard:
|
|||
ttk.Button(control_frame, text="🔄 Set to Default",
|
||||
command=self.set_to_default).grid(row=row, column=0, columnspan=2, pady=(10, 0), sticky=(tk.W, tk.E))
|
||||
|
||||
row += 1
|
||||
|
||||
# ADD EMPTY SPACER
|
||||
spacer = ttk.Label(control_frame, text="")
|
||||
spacer.grid(row=row, column=0, columnspan=2, pady=10)
|
||||
|
||||
# Add this after the parameters section
|
||||
ttk.Label(control_frame, text="Final date of growth:",
|
||||
font=('Arial', 9, 'bold')).grid(row=row, column=0, columnspan=2, sticky=tk.W, pady=(10, 4))
|
||||
|
||||
row += 1
|
||||
# Compact date entry with calendar popup
|
||||
# To get the selected dat simply call self.date_entry.get_date()
|
||||
self.date_entry = DateEntry(control_frame,
|
||||
width=12,
|
||||
background='darkblue',
|
||||
foreground='white',
|
||||
borderwidth=2,
|
||||
font=('Arial', 8),
|
||||
date_pattern='dd/mm/yyyy',
|
||||
state='readonly', # Add this
|
||||
cursor='hand2') # Add this
|
||||
self.date_entry.grid(row=row, column=0, columnspan=2, sticky=tk.W, pady=2)
|
||||
|
||||
row += 1
|
||||
|
||||
control_frame.columnconfigure(0, weight=1)
|
||||
control_frame.columnconfigure(1, weight=1)
|
||||
|
||||
|
@ -218,29 +242,57 @@ class PlantGrowthDashboard:
|
|||
command=self.submit_plant_data).pack(fill=tk.X)
|
||||
|
||||
def setup_plant_evolution_tab(self, notebook):
|
||||
"""Evolution tab"""
|
||||
"""Evolution tab with single image display"""
|
||||
evolution_frame = ttk.Frame(notebook)
|
||||
notebook.add(evolution_frame, text="🌿 Growth Evolution")
|
||||
|
||||
# Create scrollable frame for plant images
|
||||
canvas_scroll = tk.Canvas(evolution_frame)
|
||||
scrollbar = ttk.Scrollbar(evolution_frame, orient="vertical", command=canvas_scroll.yview)
|
||||
self.scrollable_frame = ttk.Frame(canvas_scroll)
|
||||
# Create main container for image display
|
||||
self.image_display_frame = ttk.Frame(evolution_frame)
|
||||
self.image_display_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10)
|
||||
|
||||
self.scrollable_frame.bind(
|
||||
"<Configure>",
|
||||
lambda e: canvas_scroll.configure(scrollregion=canvas_scroll.bbox("all"))
|
||||
# Create label for image or fallback text
|
||||
self.evolution_image_label = ttk.Label(
|
||||
self.image_display_frame,
|
||||
text="Plant state after the prediction will appear here",
|
||||
font=('Arial', 12),
|
||||
foreground='gray',
|
||||
anchor='center'
|
||||
)
|
||||
|
||||
canvas_scroll.create_window((0, 0), window=self.scrollable_frame, anchor="nw")
|
||||
canvas_scroll.configure(yscrollcommand=scrollbar.set)
|
||||
|
||||
canvas_scroll.pack(side="left", fill="both", expand=True)
|
||||
scrollbar.pack(side="right", fill="y")
|
||||
|
||||
self.image_display_frame = ttk.Frame(self.scrollable_frame)
|
||||
self.image_display_frame.pack(fill=tk.BOTH, expand=True, padx=6, pady=6)
|
||||
|
||||
self.evolution_image_label.pack(expand=True)
|
||||
|
||||
def update_evolution_image(self, filename=None):
|
||||
"""Update the evolution tab with an image from file or show fallback text"""
|
||||
if filename and os.path.exists(filename):
|
||||
try:
|
||||
# Load and display the image
|
||||
from PIL import Image, ImageTk
|
||||
|
||||
# Open and resize image if needed
|
||||
pil_image = Image.open(filename)
|
||||
# Optional: resize to fit the display area
|
||||
pil_image = pil_image.resize((400, 300), Image.Resampling.LANCZOS)
|
||||
|
||||
# Convert to PhotoImage for tkinter
|
||||
photo_image = ImageTk.PhotoImage(pil_image)
|
||||
|
||||
# Display the image
|
||||
self.evolution_image_label.config(image=photo_image, text="")
|
||||
self.evolution_image_label.image = photo_image # Keep reference to prevent garbage collection
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error loading image {filename}: {e}")
|
||||
# Show fallback text on error
|
||||
self.evolution_image_label.config(
|
||||
image="",
|
||||
text="Plant state after the prediction will appear here"
|
||||
)
|
||||
else:
|
||||
# Show fallback text when no filename or file doesn't exist
|
||||
self.evolution_image_label.config(
|
||||
image="",
|
||||
text="Plant state after the prediction will appear here"
|
||||
)
|
||||
|
||||
def setup_results_panel(self, parent):
|
||||
results_frame = ttk.LabelFrame(parent, text="Growth Prediction", padding="6")
|
||||
results_frame.grid(row=1, column=2, sticky=(tk.W, tk.E, tk.N, tk.S), padx=(6, 0))
|
||||
|
@ -249,15 +301,39 @@ class PlantGrowthDashboard:
|
|||
ttk.Label(results_frame, text="Forecast Results:",
|
||||
font=('Arial', 9, 'bold')).grid(row=0, column=0, sticky=tk.W, pady=(0, 4))
|
||||
|
||||
self.results_text = tk.Text(results_frame, height=20, width=26, wrap=tk.WORD, font=('Arial', 8))
|
||||
self.results_text = tk.Text(results_frame, height=10, width=26, wrap=tk.WORD, font=('Arial', 8), state='disabled')
|
||||
self.results_text.grid(row=1, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))
|
||||
|
||||
img = Image.open("public/TransparentFlower.png")
|
||||
|
||||
# Resize if needed
|
||||
img = img.resize((180, 290), Image.Resampling.LANCZOS)
|
||||
photo = ImageTk.PhotoImage(img)
|
||||
|
||||
static_image_label = ttk.Label(results_frame, image=photo)
|
||||
static_image_label.image = photo # Keep reference
|
||||
static_image_label.grid(row=2, column=0, pady=(6, 0))
|
||||
|
||||
results_frame.columnconfigure(0, weight=1)
|
||||
results_frame.rowconfigure(1, weight=1)
|
||||
|
||||
def on_mode_change(self):
|
||||
self.update_prediction()
|
||||
#KEY FOCUS IS TO IMPLEMENT THIS PART
|
||||
"""Handle mode changes"""
|
||||
current_mode = self.ambient_mode.get()
|
||||
|
||||
if current_mode == "controlled":
|
||||
print("Switched to Controlled mode")
|
||||
# Enable all parameter controls
|
||||
# No need to call the meteo api
|
||||
|
||||
elif current_mode == "open":
|
||||
print("Switched to Open mode")
|
||||
# Disable most parameter controls (temp, humidity, light)
|
||||
|
||||
# Call the meteo api to retrieve all the parameters, set variable meteo_values to retrieve when submitiing all
|
||||
# Inside the retrieving of all the data check if the mode is one, select which data to use
|
||||
|
||||
def on_param_change(self, param):
|
||||
value = self.env_params[param].get()
|
||||
|
||||
|
@ -269,19 +345,16 @@ class PlantGrowthDashboard:
|
|||
value_label.config(text=f"{value:.0f}")
|
||||
else:
|
||||
value_label.config(text=f"{value:.1f}")
|
||||
|
||||
# Auto-update prediction when parameters change
|
||||
self.update_prediction()
|
||||
|
||||
def set_to_default(self):
|
||||
"""Reset all parameters to default values"""
|
||||
for param_name, default_value in self.default_params.items():
|
||||
self.env_params[param_name].set(default_value)
|
||||
|
||||
self.update_parameter_label(param_name, default_value)
|
||||
|
||||
self.ambient_mode.set("controlled")
|
||||
self.update_prediction()
|
||||
|
||||
self.date_entry.set_date(date.today())
|
||||
|
||||
def update_parameter_label(self, param, value):
|
||||
"""Update the value label for a specific parameter"""
|
||||
|
@ -302,6 +375,7 @@ class PlantGrowthDashboard:
|
|||
|
||||
def load_baseline_image(self):
|
||||
self.results_text.delete(1.0, tk.END)
|
||||
self.set_to_default()
|
||||
|
||||
file_path = filedialog.askopenfilename(
|
||||
title="Select baseline plant image",
|
||||
|
@ -309,7 +383,6 @@ class PlantGrowthDashboard:
|
|||
)
|
||||
if file_path:
|
||||
self.baseline_image_path = file_path
|
||||
self.update_prediction()
|
||||
|
||||
def submit_plant_data(self):
|
||||
"""Submit plant information and photo"""
|
||||
|
@ -324,7 +397,9 @@ class PlantGrowthDashboard:
|
|||
'timestamp': datetime.now().isoformat(),
|
||||
'parameters': params,
|
||||
'baseline_image_path': self.baseline_image_path,
|
||||
'plant_info': self.plant_info_text.get(1.0, tk.END)
|
||||
'plant_info': self.plant_info_text.get(1.0, tk.END),
|
||||
'start date': datetime.now().date().isoformat(),
|
||||
'end_date': self.date_entry.get_date().isoformat()
|
||||
}
|
||||
|
||||
#Remove plant_info_text
|
||||
|
@ -339,34 +414,35 @@ class PlantGrowthDashboard:
|
|||
json.dump(submission_data, f, indent=4)
|
||||
|
||||
# Here call the bot pipeline to store results on files in plant_data
|
||||
# results are in the form of (text, image)
|
||||
results = "come bot please"
|
||||
|
||||
text = getattr(results, 'text', None)
|
||||
image_filename = getattr(results, 'image', None)
|
||||
images_dir = "./plant_data"
|
||||
os.makedirs(data_dir, exist_ok=True)
|
||||
image_path = os.path.join(images_dir, image_filename)
|
||||
self.updating_evolution_and_forecasts(text, image_path)
|
||||
|
||||
# Here update the informations in the last box from plant_data/texts
|
||||
|
||||
# Here update the informations in growth evolution from plant_data/images
|
||||
|
||||
#Calling a small advertment to notify the user that the image has been generated
|
||||
messagebox.showinfo("Submission successful, go to growth evolution to see the results")
|
||||
print(f"Submission data saved to: {filepath}")
|
||||
except Exception as e:
|
||||
messagebox.showerror("Submission Error", f"Error submitting data: {str(e)}")
|
||||
|
||||
def update_prediction(self):
|
||||
try:
|
||||
# Get current parameters
|
||||
params = {param: var.get() for param, var in self.env_params.items()}
|
||||
params['plant_type'] = self.current_plant
|
||||
params['ambient_mode'] = self.ambient_mode.get()
|
||||
|
||||
# Generate prediction
|
||||
prediction = self.plant_model.predict_growth(params)
|
||||
|
||||
# Update initial plant display
|
||||
self.update_initial_plant_display()
|
||||
|
||||
# Update results text
|
||||
self.update_results_display(prediction)
|
||||
|
||||
except Exception as e:
|
||||
messagebox.showerror("Prediction Error", f"Error generating prediction: {str(e)}")
|
||||
|
||||
|
||||
def updating_evolution_and_forecasts(self, text, image_path):
|
||||
self.results_text.config(state='normal') # Enable editing
|
||||
self.results_text.delete(1.0, tk.END) # Clear existing content
|
||||
if text != None:
|
||||
self.results_text.insert(1.0, text) # Insert new text
|
||||
self.results_text.config(state='disabled') # Disable editing again
|
||||
|
||||
self.update_evolution_image(image_path)
|
||||
|
||||
def update_initial_plant_display(self):
|
||||
"""Update the initial plant state display"""
|
||||
try:
|
||||
|
|
25
path.txt
25
path.txt
|
@ -1,35 +1,18 @@
|
|||
ROADMAP
|
||||
-understand codebase
|
||||
|
||||
-add the possibility to retrieve all the data when the user click the submit button
|
||||
-format the data in an usable format
|
||||
|
||||
-differences between the three modes:
|
||||
Open mode:
|
||||
-the temp, humidity, light, water parameters are setted by the meteo api, the rest is setted by the user
|
||||
|
||||
-all the parameters are controlled by the user
|
||||
SemiControlled mode:
|
||||
-the user choose how to set the parameters
|
||||
-all parameters free
|
||||
|
||||
Controlled mode:
|
||||
-all the values are set by the user
|
||||
|
||||
|
||||
-default modes are setted with general values in an optimal condition for your plant
|
||||
-the user choose which parameters are free and which are controlled by the meteo api
|
||||
|
||||
-The forecast results will have as input the text description obtained by the model
|
||||
(the output will be in the text folder inside data/texts folder, format "date1-date2-enum.txt")
|
||||
-The growth evolution will have as input the photo/s obtained by the model
|
||||
(the output will be in the text folder inside data/images folder, format "date1-date2-enum.jpg)
|
||||
|
||||
-set the value of light as Daily Light Integral (DLI)
|
||||
|
||||
-forecast for a period of time:
|
||||
-create a calendar widget with start and end data
|
||||
-based on how much does it take to create an image, for each subperiod of time create a new image to show
|
||||
(it this will be coded add all a button to iterate over the folder)
|
||||
|
||||
-when the user click the load plant image, all previous record will be eliminated
|
||||
-make the calendar widget working
|
||||
|
||||
-final updates of README.md, hackathon page, forgejo, github page, small design adjustments.
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue