update_

main.py

@@ -66,10 +66,13 @@ try:
     gensim_data_dir = os.path.join(cache_dir, 'gensim-data')
     upload_dir = os.path.join(cache_dir, 'uploads')
     ans_image_dir = os.path.join(cache_dir, 'ans_image')
+    htr_images_dir = os.path.join(cache_dir, 'images')
 
     # Create directories with correct permissions
-    for directory in [cache_dir, nltk_data_dir, gensim_data_dir, upload_dir, ans_image_dir]:
+    for directory in [cache_dir, nltk_data_dir, gensim_data_dir, upload_dir, ans_image_dir, htr_images_dir]:
         os.makedirs(directory, exist_ok=True)
+        # Ensure directory has write permissions
+        os.chmod(directory, 0o777)
 except Exception as e:
     print(f"Warning: Could not create cache directories: {e}")
     # Fallback to temporary directories
@@ -78,6 +81,7 @@ except Exception as e:
     gensim_data_dir = tempfile.mkdtemp(prefix='gensim_data_')
     upload_dir = tempfile.mkdtemp(prefix='uploads_')
     ans_image_dir = tempfile.mkdtemp(prefix='ans_image_')
+    htr_images_dir = tempfile.mkdtemp(prefix='htr_images_')
 
 # Set environment variables
 os.environ['HF_HOME'] = cache_dir
@@ -280,6 +284,10 @@ def compute_marks():
                 processed_answers.append(ans)
         answers = processed_answers
 
+        # Create HTR images directory if it doesn't exist
+        os.makedirs(htr_images_dir, exist_ok=True)
+        os.chmod(htr_images_dir, 0o777)  # Ensure write permissions
+
         # Get files from the request
         files = request.files.getlist('files[]')
         if not files:
@@ -299,16 +307,24 @@ def compute_marks():
                     # Create student directory if it doesn't exist
                     student_dir = os.path.join(ans_image_dir, student_folder)
                     os.makedirs(student_dir, exist_ok=True)
+                    os.chmod(student_dir, 0o777)  # Ensure write permissions
                     
                     # Save the file
                     filepath = os.path.join(student_dir, filename)
                     file.save(filepath)
+                    os.chmod(filepath, 0o666)  # Ensure file is readable/writable
+                    
+                    # Create symlink in HTR images directory
+                    htr_filepath = os.path.join(htr_images_dir, f"{student_folder}_{filename}")
+                    if os.path.exists(htr_filepath):
+                        os.remove(htr_filepath)
+                    os.symlink(filepath, htr_filepath)
                     
                     # Add to data structure
                     if student_folder not in data:
                         data[student_folder] = []
-                    data[student_folder].append(filepath.replace("\\", "/"))
-                    log_print(f"Saved file: {filepath}")
+                    data[student_folder].append(htr_filepath.replace("\\", "/"))
+                    log_print(f"Saved file: {filepath} and linked to {htr_filepath}")
 
         if not data:
             log_print("No valid image files were found in the upload", "ERROR")
@@ -439,11 +455,12 @@ def allowed_file(filename, allowed_extensions):
 def cleanup_temp_files():
     try:
         import shutil
-        temp_dirs = [ans_image_dir, upload_dir]
+        temp_dirs = [ans_image_dir, upload_dir, htr_images_dir]
         for directory in temp_dirs:
             if os.path.exists(directory):
                 shutil.rmtree(directory)
                 os.makedirs(directory, exist_ok=True)
+                os.chmod(directory, 0o777)  # Ensure write permissions
         log_print("Successfully cleaned up temporary files")
     except Exception as e:
         log_print(f"Error cleaning up temporary files: {e}", "ERROR")