jjz5463
commited on
Commit
·
49671af
1
Parent(s):
868e527
GUI update, hide some files
Browse files
Experiments/Baseline/GUI.py
CHANGED
|
@@ -5,6 +5,7 @@ from PIL import Image
|
|
| 5 |
from google.oauth2 import service_account
|
| 6 |
from baseline_utils import detect_text_in_image, summarize_diary_text, analyze_writer_image, generate_comic_book
|
| 7 |
import glob
|
|
|
|
| 8 |
|
| 9 |
# Load secrets
|
| 10 |
openai_api_key = st.secrets["general"]["openai_api_key"]
|
|
@@ -30,9 +31,9 @@ if uploaded_diary and uploaded_writer_image:
|
|
| 30 |
writer_image = Image.open(uploaded_writer_image)
|
| 31 |
|
| 32 |
# Save the file-like objects as image files (optional if needed)
|
| 33 |
-
diary_image_path = "temp_diary_image.png"
|
| 34 |
-
writer_image_path = "temp_writer_image.png"
|
| 35 |
-
|
| 36 |
diary_image.save(diary_image_path)
|
| 37 |
writer_image.save(writer_image_path)
|
| 38 |
|
|
|
|
| 5 |
from google.oauth2 import service_account
|
| 6 |
from baseline_utils import detect_text_in_image, summarize_diary_text, analyze_writer_image, generate_comic_book
|
| 7 |
import glob
|
| 8 |
+
import os
|
| 9 |
|
| 10 |
# Load secrets
|
| 11 |
openai_api_key = st.secrets["general"]["openai_api_key"]
|
|
|
|
| 31 |
writer_image = Image.open(uploaded_writer_image)
|
| 32 |
|
| 33 |
# Save the file-like objects as image files (optional if needed)
|
| 34 |
+
diary_image_path = "temp_upload_images/temp_diary_image.png"
|
| 35 |
+
writer_image_path = "temp_upload_images/temp_writer_image.png"
|
| 36 |
+
os.makedirs("temp_upload_images", exist_ok=True)
|
| 37 |
diary_image.save(diary_image_path)
|
| 38 |
writer_image.save(writer_image_path)
|
| 39 |
|
Experiments/Baseline/input_images/test_sample.jpeg
DELETED
|
Binary file (168 kB)
|
|
|
Experiments/Baseline/input_images/writer.jpeg
DELETED
|
Binary file (67.1 kB)
|
|
|