Update app.py

app.py

@@ -1620,38 +1620,12 @@ def process_image_old_05152024(image_input):
         )
         st.markdown(response.choices[0].message.content)
 
-def save_image(image_input):
+def save_image(image_input, filename_txt):
     # Save the uploaded video file
-    with open(image_input.name, "wb") as f:
+    with open(filename_txt, "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")
-        filename_parts = []
-        for line in lines:
-            words = re.findall(r'\b\w+\b', line)
-            alphanumeric_words = [word for word in words if word.isalnum()]
-            if len(alphanumeric_words) >= 2:
-                filename_parts.append(f"{alphanumeric_words[0]}_{alphanumeric_words[1]}")
-
-        # 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")
@@ -1670,10 +1644,16 @@ def process_image(image_input):
         )
         image_response = response.choices[0].message.content
         st.markdown(image_response)
+        create_file()
 
         # Save the image with a new filename based on the response
-        #save_image(image_input, image_response)        
-        save_image(image_input)        
+        #save_image(image_input, image_response)   
+        
+        filename_txt = generate_filename(image_response, "md")
+        create_file(filename_txt, transcript, response, should_save)
+
+        filename_txt = generate_filename(image_response, "png")
+        save_image(image_input, filename_txt)        
 
 def process_audio(audio_input):
     if audio_input: