update code and fix bug

Author: muzammaldeveloper

.gitignore

@@ -1,31 +1,33 @@
-# -----------------------------
-# Global
-# -----------------------------
+# ==============================
+# OS Files
+# ==============================
 .DS_Store
 Thumbs.db
 Desktop.ini
+ehthumbs.db
+
+# ==============================
+# IDE / Editor
+# ==============================
 .vscode/
 .idea/
-*.log
+*.swp
+*.swo
 *.tmp
 *.temp
 *.bak
 *.orig
 *.rej
 
-# Local scratch docs / notes
-PROJECT_DETAILS.txt
-
-# -----------------------------
+# ==============================
 # Secrets / Environment
-# -----------------------------
+# ==============================
 **/.env
-**/.env.*
 !**/.env.example
 
-# -----------------------------
+# ==============================
 # Python
-# -----------------------------
+# ==============================
 **/__pycache__/
 **/*.py[cod]
 **/*.pyo
@@ -34,27 +36,45 @@ PROJECT_DETAILS.txt
 **/.coverage
 **/htmlcov/
 **/.ruff_cache/
+**/.mypy_cache/
+**/.hypothesis/
+**/dist/
+**/build/
+**/*.egg
+**/.eggs/
 **/venv/
 **/.venv/
 **/env/
 
-# SQLite / local db
+# SQLite database
 **/*.db
 **/*.sqlite3
+Backend/instance/
 
-# Celery runtime
+# Celery
 **/celerybeat-schedule
 **/celerybeat.pid
 
-# Backend generated storage artifacts
+# Migrations auto-generated
+Backend/migrations/versions/*.py
+
+# ==============================
+# Generated Storage Files
+# ==============================
 Backend/storage/images/*
 Backend/storage/documents/*
 !Backend/storage/images/.gitkeep
 !Backend/storage/documents/.gitkeep
 
-# -----------------------------
-# Node / Next.js
-# -----------------------------
+# ==============================
+# Logs
+# ==============================
+*.log
+logs/
+
+# ==============================
+# Node / Next.js / Frontend
+# ==============================
 Frontend/node_modules/
 Frontend/.next/
 Frontend/out/
@@ -67,7 +87,17 @@ Frontend/yarn-debug.log*
 Frontend/yarn-error.log*
 Frontend/pnpm-debug.log*
 
-# Optional package manager locks (uncomment per team preference)
-# Frontend/package-lock.json
-# Frontend/yarn.lock
-# Frontend/pnpm-lock.yaml
+# ==============================
+# Docker (local only)
+# ==============================
+# Uncomment if you don't want docker files on GitHub
+# Backend/Dockerfile
+# Backend/docker-compose.yml
+
+# ==============================
+# Misc
+# ==============================
+PROJECT_DETAILS.txt
+*.zip
+*.tar.gz
+Backend/package-lock.json

Backend/README.md

@@ -1,2 +1,8 @@
 """AI Assignment Generator — v2.0"""
-__version__ = "2.0.0"
+
+from app.factory import create_app
+
+__version__ = "2.0.0"
+
+# Create app instance for Flask to discover
+app = create_app()

Backend/app/__init__.py

@@ -85,6 +85,9 @@ def generate_assignment():
         word_count=data["word_count"],
         citation_style=data["citation_style"],
         template=data["template"],
+        student_name=data.get("student_name", ""),
+        roll_number=data.get("roll_number", ""),
+        department=data.get("department", ""),
         status=AssignmentStatus.PENDING.value,
         progress_percent=0,
     )
@@ -205,6 +208,31 @@ def download_assignment(assignment_id: str):
     return send_file(abs_path, mimetype=mimetype, as_attachment=True, download_name=name)
 
 
+@assignment_bp.route("/<string:assignment_id>/update-info", methods=["PATCH"])
+@jwt_required()
+def update_assignment_info(assignment_id: str):
+    """Update student name and roll number."""
+    user_id = get_jwt_identity()
+    assignment = db.session.get(Assignment, assignment_id)
+
+    if not assignment:
+        return jsonify({"success": False, "message": "Assignment not found."}), 404
+    if assignment.user_id != user_id:
+        return jsonify({"success": False, "message": "Access denied."}), 403
+
+    import bleach
+    data = request.get_json() or {}
+    if "student_name" in data:
+        assignment.student_name = bleach.clean(str(data["student_name"])[:200], tags=[], strip=True)
+    if "roll_number" in data:
+        assignment.roll_number = bleach.clean(str(data["roll_number"])[:100], tags=[], strip=True)
+    if "department" in data:
+        assignment.department = bleach.clean(str(data["department"])[:200], tags=[], strip=True)
+
+    db.session.commit()
+    return jsonify({"success": True, "message": "Info updated."}), 200
+
+
 @assignment_bp.route("/<string:assignment_id>/sections/<string:section_id>", methods=["PATCH"])
 @jwt_required()
 def update_section(assignment_id: str, section_id: str):
@@ -289,11 +317,11 @@ def regenerate_documents(assignment_id: str):
         # Gather existing images
         images = [
             GeneratedImage(
-                section_title=img.caption.split(": ", 1)[-1] if ": " in img.caption else "",
-                image_path=os.path.abspath(img.image_url) if img.image_url else "",
-                caption=img.caption,
+                section_title=img.caption.split(": ", 1)[-1] if img.caption and ": " in img.caption else "",
+                image_path=img.image_url if img.image_url and os.path.exists(img.image_url) else "",
+                caption=img.caption or "",
                 prompt=img.prompt or "",
-                success=bool(img.image_url and os.path.exists(os.path.abspath(img.image_url))),
+                success=bool(img.image_url and os.path.exists(img.image_url)),
             )
             for img in assignment.images
         ]
@@ -305,10 +333,16 @@ def regenerate_documents(assignment_id: str):
         docx_path = doc_gen.generate_docx(
             content=content, images=images,
             template=assignment.template, assignment_id=assignment.id,
+            student_name=assignment.student_name or "",
+            roll_number=assignment.roll_number or "",
+            department=assignment.department or "",
         )
         pdf_path = doc_gen.generate_pdf(
             content=content, images=images,
             template=assignment.template, assignment_id=assignment.id,
+            student_name=assignment.student_name or "",
+            roll_number=assignment.roll_number or "",
+            department=assignment.department or "",
         )
 
         assignment.docx_path = docx_path