Update app.py

app.py

@@ -1540,10 +1540,19 @@ def create_file(filename, prompt, response, is_image=False):
     with open(filename, "w", encoding="utf-8") as f:
         f.write(prompt + "\n\n" + response)
 
-def save_image(image, filename):
+def save_image_old2(image, filename):
     with open(filename, "wb") as f:
         f.write(image.getbuffer())
 
+# Now filename length protected for linux and windows filename lengths
+def save_image(image, filename):
+    max_filename_length = min(os.pathconf('/', 'PC_NAME_MAX'), 255, 260)  # Linux: 255, Windows: 260 (including path)
+    filename_stem, extension = os.path.splitext(filename)
+    truncated_stem = filename_stem[:max_filename_length - len(extension)] if len(filename) > max_filename_length else filename_stem
+    filename = f"{truncated_stem}{extension}"
+    with open(filename, "wb") as f:
+        f.write(image.getbuffer())
+        
 def extract_boldface_terms(text):
     return re.findall(r'\*\*(.*?)\*\*', text)