sosokker/Packaged-Food-Explorer
1
0
Fork 0
mirror of https://github.com/Sosokker/Packaged-Food-Explorer.git synced 2025-12-19 05:04:06 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix embed function + Add subplot pie/bar

Browse Source
This commit is contained in:
Sirin Puenggun 2023-05-11 23:31:25 +07:00
parent aeb391eabf
commit 914c92ea4d
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
Essential/plotter.py
View File

@ -1,4 +1,5 @@
import os import os
from plotly.subplots import make_subplots
import plotly.graph_objects as go import plotly.graph_objects as go
import plotly.io as pio import plotly.io as pio
import tempfile import tempfile
@ -35,13 +36,20 @@ class plotter:
if chart_type == 'bar': if chart_type == 'bar':
fig = go.Figure(data=go.Bar(x=nutrient_names, y=nutrient_values, marker_color='skyblue')) fig = go.Figure(data=go.Bar(x=nutrient_names, y=nutrient_values, marker_color='skyblue'))
fig.update_layout(xaxis_title='Nutrients', fig.update_layout(title='Macronutrients Graph',xaxis_title='Nutrients',
yaxis_title='Value') yaxis_title='Value')
elif chart_type == 'pie': elif chart_type == 'pie':
fig = go.Figure(data=go.Pie(labels=nutrient_names, values=nutrient_values)) fig = go.Figure(data=go.Pie(labels=nutrient_names, values=nutrient_values))
# fig.update_layout(title=f'Nutrient of {product_name}') fig.update_layout(title=f'Nutrient of {product_name}')
elif chart_type == 'barpie':
fig = make_subplots(rows=2, cols=2, specs=[[{"type": "xy"}, {"type": "domain"}], [{}, {}]])
fig.add_trace(go.Pie(labels=nutrient_names, values=nutrient_values),
row=1, col=2)
fig.add_trace(go.Bar(x=nutrient_names, y=nutrient_values, marker_color='skyblue'),
row=1, col=1)
fig.update_layout(title=f'Nutrient of {product_name}')
else: else:
raise ValueError('Invalid chart type. Please choose either "bar" or "pie".') raise ValueError('Invalid chart type. Please choose either "bar" or "pie" or "barpie".')
html_string = pio.to_html(fig, include_plotlyjs='cdn', full_html=False) html_string = pio.to_html(fig, include_plotlyjs='cdn', full_html=False)
encoded_html = html_string.encode('utf-8') encoded_html = html_string.encode('utf-8')
@ -49,11 +57,14 @@ class plotter:
if popup: if popup:
with tempfile.NamedTemporaryFile(suffix=".html", delete=False) as temp: with tempfile.NamedTemporaryFile(suffix=".html", delete=False) as temp:
filename = temp.name filename = temp.name
print(filename)
temp.write(encoded_html) temp.write(encoded_html)
webview.create_window('Nutrients Graph', filename) temp.close()
webview.start() webview.create_window('Nutrients Graph', filename)
webview.start()
os.remove(filename)
else: else:
fig.update_layout(width=250, height=300) fig.update_layout(width=350, height=300)
temp_dir = tempfile.mkdtemp() temp_dir = tempfile.mkdtemp()
image_path = os.path.join(temp_dir, "plot.jpeg") image_path = os.path.join(temp_dir, "plot.jpeg")
fig.write_image(image_path, format="jpeg") fig.write_image(image_path, format="jpeg")
@ -62,5 +73,6 @@ class plotter:
label.img = ImageTk.PhotoImage(img) label.img = ImageTk.PhotoImage(img)
label['image'] = label.img label['image'] = label.img
label.pack() label.pack()
os.remove(image_path)
# print(type(nutrient_plotter(df, 1, [16, 18, 20, 22], 'bar', False, True))) # print(type(nutrient_plotter(df, 1, [16, 18, 20, 22], 'bar', False, True)))