Update app.py

app.py

@@ -1621,7 +1621,30 @@ def process_image_old_05152024(image_input):
         st.markdown(response.choices[0].message.content)
 
 
+
 def save_image(image_input, image_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 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")