feat: schengen

app/util/{JapanMultiEntryvisaLetterGenerator.py → japan_multientry_visa_letter_generator.py}

@@ -1,6 +1,6 @@
 import datetime
 from fpdf import FPDF
-from .PDFDocumentGenerator import PDFDocumentGenerator # The fpdf2-based one
+from .pdf_document_generator import PDFDocumentGenerator # The fpdf2-based one
 
 class JapanMultiEntryVisaLetterGenerator(PDFDocumentGenerator):
     """
@@ -26,7 +26,6 @@ class JapanMultiEntryVisaLetterGenerator(PDFDocumentGenerator):
         formatted_date = today.strftime("%d, %B, %Y")
 
         pdf.set_font('Arial', '', 11)
-        pdf.set_margins(25, 25)
         pdf.set_y(25)
         pdf.set_left_margin(25)
 

app/util/{PDFDocumentGenerator.py → pdf_document_generator.py}

@@ -17,6 +17,7 @@ class PDFDocumentGenerator(ABC):
         """
         self.data = data
 
+
     @abstractmethod
     def build_document(self, pdf: FPDF):
         """
@@ -37,24 +38,20 @@ class PDFDocumentGenerator(ABC):
                  On failure, (None, error_message).
         """
         try:
-            # 1. Initialize the FPDF object
             pdf = FPDF()
             pdf.add_page()
             
-            # 2. Let the subclass build the document
             self.build_document(pdf)
 
-            # 3. Define file path and save the document
             pdf_filename = "document.pdf"
             pdf_filepath = os.path.join(tempdir, pdf_filename)
             
             pdf.output(pdf_filepath)
             
-            # 4. Check if PDF was successfully created
             if not os.path.exists(pdf_filepath):
                 return None, "PDF file was not created by fpdf2."
 
-            # Success
+            pdf.set_margins(25, 25)
             return pdf_filepath, None
 
         except Exception as e:

app/util/schengen_visa_letter_generator.py

@@ -0,0 +1,143 @@
+import datetime
+from fpdf import FPDF
+from .pdf_document_generator import PDFDocumentGenerator
+
+
+class SchengenVisaLetterGenerator(PDFDocumentGenerator):
+    """
+    Generates a Schengen Visa Cover Letter PDF (Individual or Group)
+    using the shared PDFDocumentGenerator base for layout consistency.
+    """
+
+    def __init__(self, data: dict, trip_type: str = "individual"):
+        super().__init__(data)
+        self.trip_type = trip_type.lower()
+
+    def build_document(self, pdf):
+        """
+        Builds the formatted PDF content.
+        This method is called by the base class `compile_pdf()`.
+        """
+        personal = self.data.get("personal_details", {})
+        members = self.data.get("group_members", [])
+
+        # --- Extract fields ---
+        city = self.data.get("city", "Jakarta")
+        date = self.data.get("date", "18th July 2024")
+        country = self.data.get("country", "[Schengen Country Name]")
+        purpose = self.data.get("purpose", "[tourism/family visit]")
+        main_dest = self.data.get("main_destination", "[country]")
+        event = self.data.get("event", "[event/activities]")
+        other_dest = self.data.get("other_destinations", "")
+        start = self.data.get("travel_start", "[start date]")
+        end = self.data.get("travel_end", "[end date]")
+        duration = self.data.get("duration", "[X days and Y nights]")
+        trip_highlight = self.data.get("trip_highlight", "[trip highlight]")
+        contact_email = self.data.get("contact_email", "[email]")
+        contact_phone = self.data.get("contact_phone", "[phone number]")
+        job_commitment = self.data.get("job_commitment", "[your job title or role]")
+        financial_status = self.data.get("financial_status", "sound")
+
+        pdf.set_font("Times", "B", 14)
+        # title = (
+        #     "COVER LETTER FOR GROUP TRIP"
+        #     if self.trip_type == "group"
+        #     else "COVER LETTER FOR INDIVIDUAL TRIP"
+        # )
+        # pdf.cell(0, 10, title, ln=True, align="C")
+        # pdf.ln(10)
+
+        pdf.set_font("Times", "", 12)
+        pdf.multi_cell(0, 8, f"{city}, {date}")
+        pdf.multi_cell(0, 8, f"To: Embassy of {country} Jakarta")
+        pdf.multi_cell(0, 8, "Subject: Application for Schengen Visa")
+        pdf.ln(6)
+
+        # --- Opening Paragraph ---
+        body = (
+            f"Dear Sir/Madam,\n\n"
+            f"I am writing to apply for a Schengen {purpose} visa for the purpose of visiting {main_dest} on {event}. "
+        )
+        if other_dest:
+            body += f"Following this, I plan to continue my journey to {other_dest}. "
+        body += (
+            f"My travel period is from {start} to {end}, covering a total of {duration}."
+        )
+        pdf.multi_cell(0, 8, body)
+        pdf.ln(8)
+
+        # --- Personal Details ---
+        pdf.set_font("Times", "B", 12)
+        pdf.cell(0, 10, "Personal Details:", ln=True)
+        pdf.set_font("Times", "", 12)
+        pdf.multi_cell(
+            0,
+            8,
+            f"Name\t\t: {personal.get('name', '')}\n"
+            f"Date of Birth\t: {personal.get('dob', '')}\n"
+            f"Nationality\t: {personal.get('nationality', '')}\n"
+            f"Occupation\t: {personal.get('occupation', '')}\n"
+            f"Passport Number\t: {personal.get('passport_number', '')}",
+        )
+        pdf.ln(6)
+
+        # --- Group Section ---
+        if self.trip_type == "group" and members:
+            pdf.multi_cell(0, 8, f"I will be attending the {event} with my family members:\n")
+            for m in members:
+                pdf.multi_cell(
+                    0,
+                    8,
+                    f"Relationship\t: {m.get('relationship', '')}\n"
+                    f"Name\t\t: {m.get('name', '')}\n"
+                    f"Date of Birth\t: {m.get('dob', '')}\n"
+                    f"Occupation\t: {m.get('occupation', '')}\n"
+                    f"Nationality\t: {m.get('nationality', '')}\n"
+                    f"Passport Number\t: {m.get('passport_number', '')}\n",
+                )
+                pdf.ln(2)
+            pdf.multi_cell(
+                0,
+                8,
+                f"We will travel together to {trip_highlight}, with a detailed itinerary attached.\n",
+            )
+            pdf.ln(6)
+
+        # --- Supporting Documents ---
+        pdf.set_font("Times", "B", 12)
+        pdf.cell(0, 10, "We will also provide the following supporting documents:", ln=True)
+        pdf.set_font("Times", "", 12)
+        pdf.multi_cell(
+            0,
+            8,
+            "- Completed visa application form\n"
+            "- Recent photographs\n"
+            "- Copy of passport\n"
+            "- Proof of travel insurance\n"
+            "- Round-trip flight itinerary\n"
+            "- Hotel accommodation reservation\n"
+            "- Bank statements\n"
+            "- Statement letter and proof of current employment\n"
+            "- Copy of family documents or marriage certificate\n"
+            "- Copy of passport",
+        )
+        pdf.ln(6)
+
+        # --- Commitment & Return ---
+        pdf.multi_cell(
+            0,
+            8,
+            f"Commitment and Return: I assure you that I/we will adhere to all Schengen visa regulations and return to Indonesia as planned. "
+            f"I am a {job_commitment}, committed to my duties in Indonesia. "
+            f"My financial status is {financial_status}, and sufficient proof of funds is attached to cover all expenses for the trip. "
+            f"I guarantee our prompt return upon completion of the journey.\n\n"
+            f"We would be grateful if you could consider our application favorably and issue the necessary visa for the trip. "
+            f"Should you require any further details, please contact me at {contact_email} or {contact_phone}.",
+        )
+        pdf.ln(8)
+
+        # --- Closing ---
+        pdf.multi_cell(0, 8, "Thank you for your time and kind attention.\n\nBest regards,\n\n")
+        pdf.multi_cell(0, 8, personal.get("name", ""))
+
+        return pdf

server.py

@@ -8,12 +8,12 @@ from dotenv import load_dotenv
 import json
 import requests
 import uuid
-import asyncio
+import importlib
 import io
 
 from app.util.gen_ai_base import GenAIBaseClient
 from app.util.browser_agent import BrowserAgent
-from app.util.JapanMultiEntryvisaLetterGenerator import JapanMultiEntryVisaLetterGenerator
+from app.util.japan_multientry_visa_letter_generator   import JapanMultiEntryVisaLetterGenerator
 import sys
 sys.stdout.reconfigure(line_buffering=True)
 API = "https://api-dev.spun.global"
@@ -112,25 +112,43 @@ def create_app() -> Flask:
     #             else:
     #                 return jsonify({"error": "Failed to retrieve visa information"}), 404
     #     return jsonify({"error": "Unexpected error"}), 500
-    @app.route('/generate/japan-multientry-tourist', methods=['POST'])
-    def generate_japan_multientry_tourist():
+    
+    @app.route('/generate/<visa_type>', methods=['POST'])
+    def generate_visa_letter(visa_type):
         """
-        Generates a Japan Visa Letter PDF and returns it as a downloadable file.
-        Expects JSON payload:
-        {
-            "name": "Your Name",
-            "address": "Your Address Line 1",
-            "city_postal": "City, Postal Code",
-            "email": "your.email@example.com",
-            "phone": "+123456789"
-        }
+        Dynamically generates visa letters based on <visa_type>.
+        Example:
+        POST /generate/schengen
+        POST /generate/japan-multientry-tourist
         """
         try:
             data = request.get_json()
             if not data:
                 return jsonify({"error": "No JSON payload provided"}), 400
 
-            letter_generator = JapanMultiEntryVisaLetterGenerator(data)
+            # ✅ Map visa types to their module and class
+            generator_map = {
+                "japan-multientry-tourist": (
+                    "app.util.japan_multientry_visa_letter_generator",
+                    "JapanMultiEntryVisaLetterGenerator",
+                ),
+                "schengen": (
+                    "app.util.schengen_visa_letter_generator",
+                    "SchengenVisaLetterGenerator",
+                ),
+            }
+
+            if visa_type.lower() not in generator_map:
+                return jsonify({"error": f"Unsupported visa type: {visa_type}"}), 400
+
+            module_path, class_name = generator_map[visa_type.lower()]
+
+            # ✅ Import dynamically using importlib
+            module = importlib.import_module(module_path)
+            generator_class = getattr(module, class_name)
+
+            # ✅ Generate the PDF
+            letter_generator = generator_class(data)
 
             with tempfile.TemporaryDirectory() as tempdir:
                 pdf_filepath, error_msg = letter_generator.compile_pdf(tempdir)
@@ -147,11 +165,11 @@ def create_app() -> Flask:
                     pdf_bytes,
                     mimetype="application/pdf",
                     as_attachment=True,
-                    download_name="Japan_Visa_Application_Letter.pdf"
+                    download_name=f"{visa_type.replace('-', '_')}_visa_letter.pdf",
                 )
 
         except Exception as e:
-            print(f"Error in route: {e}")
+            print(f"Error in /generate/{visa_type}: {e}")
             return jsonify({"error": str(e)}), 500
 
     @app.route('/', methods=['GET'])