Update app.py

app.py

@@ -1620,9 +1620,18 @@ def process_image_old_05152024(image_input):
         )
         st.markdown(response.choices[0].message.content)
 
-
+def save_image(image_input):
+    # Save the uploaded video file
+    with open(image_input.name, "wb") as f:
+        f.write(image_input.getbuffer())
+    return image_input.name
 
 def save_image(image_input, image_response):
+    if image_input is not None:
+        # Save the uploaded video file
+        image_path = save_image(image_input)
+        return image_path  # finish below code where it infers file naming from response
+    
     if image_input and image_response:
         # Extract the first two alphanumeric words from each line
         lines = image_response.split("\n")
@@ -1643,29 +1652,6 @@ def save_image(image_input, image_response):
 
         return filename
 
-
-def save_image_old(image_input, image_response):
-    if image_input and image_response:
-        # Extract emojis and first two words from each markdown line
-        lines = image_response.split("\n")
-        filename_parts = []
-        for line in lines:
-            if line.startswith("- "):
-                emoji = re.search(r'(\p{Extended_Pictographic})', line)
-                words = re.findall(r'\b\w+\b', line)
-                if emoji and len(words) >= 2:
-                    filename_parts.append(f"{emoji.group(1)}_{words[1]}_{words[2]}")
-
-        # Create the filename by concatenating the extracted parts
-        filename = "_".join(filename_parts)[:50]  # Limit filename length to 50 characters
-        filename = f"{filename}.png"
-
-        # Save the image with the new filename
-        with open(filename, "wb") as f:
-            f.write(image_input.getbuffer())
-
-        return filename
-
 def process_image(image_input):
     if image_input:
         base64_image = base64.b64encode(image_input.read()).decode("utf-8")
@@ -1790,6 +1776,9 @@ def process_audio_and_video(video_input):
 
 
 def main():
+
+
+    
     st.markdown("### OpenAI GPT-4o Model")
     st.markdown("#### The Omni Model with Text, Audio, Image, and Video")
     option = st.selectbox("Select an option", ("Text", "Image", "Audio", "Video"))
@@ -1805,14 +1794,22 @@ def main():
         video_input = st.file_uploader("Upload a video file", type=["mp4"])
         process_audio_and_video(video_input)
 
+    # Image and Video Galleries
+    num_columns_video=st.slider(key="num_columns_video", label="Choose Number of Video Columns", min_value=1, max_value=15, value=4)
+    display_videos_and_links(num_columns_video)   # Video Jump Grid
+    num_columns_images=st.slider(key="num_columns_images", label="Choose Number of Image Columns", min_value=1, max_value=15, value=4)
+    display_images_and_wikipedia_summaries(num_columns_images)   # Image Jump Grid
+
+
+
+
+    
+
 if __name__ == "__main__":
     main()
         
 display_glossary_grid(roleplaying_glossary)  # Word Glossary Jump Grid - Dynamically calculates columns based on details length to keep topic together
-num_columns_video=st.slider(key="num_columns_video", label="Choose Number of Video Columns", min_value=1, max_value=15, value=4)
-display_videos_and_links(num_columns_video)   # Video Jump Grid
-num_columns_images=st.slider(key="num_columns_images", label="Choose Number of Image Columns", min_value=1, max_value=15, value=4)
-display_images_and_wikipedia_summaries(num_columns_images)   # Image Jump Grid
+
 num_columns_text=st.slider(key="num_columns_text", label="Choose Number of Text Columns", min_value=1, max_value=15, value=4)
 display_buttons_with_scores(num_columns_text)  # Feedback Jump Grid