Update app.py

app.py

@@ -59,39 +59,42 @@ def generate_itinerary(mood, location, budget, days):
     image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
     return itinerary_text, image_path, gr.update(visible=True)
 
-def generate_pdf(itinerary_text):
+def generate_pdf(itinerary_text, location="Destination", days=5):
     if not itinerary_text.strip():
         return None
 
-    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-    c = canvas.Canvas(temp_pdf.name, pagesize=letter)
-    width, height = letter
-
-    margin = 40
-    line_height = 12
-    font_size = 10
-    c.setFont("Courier", font_size)
-
-    max_chars_per_line = int((width - 2 * margin) / (font_size * 0.6))
-
-    x = margin
-    y = height - margin
-
-    lines = []
-    for paragraph in itinerary_text.split('\n'):
-        wrapped = wrap(paragraph, width=max_chars_per_line)
-        lines.extend(wrapped if wrapped else [''])
-
-    for line in lines:
-        if y < margin + line_height:
-            c.showPage()
-            c.setFont("Courier", font_size)
-            y = height - margin
-        c.drawString(x, y, line)
-        y -= line_height
-
-    c.save()
-    return temp_pdf.name
+    # Create a nice filename
+    safe_location = re.sub(r'[^A-Za-z0-9]+', '_', location.strip())
+    filename = f"itinerary_{safe_location}_{days}_days.pdf"
+    file_path = os.path.join(tempfile.gettempdir(), filename)
+
+    # Prepare the document template with letter page size
+    doc = SimpleDocTemplate(file_path, pagesize=letter,rightMargin=40, leftMargin=40,topMargin=40, bottomMargin=40)
+    # Get default stylesheet and customize the Normal style
+    styles = getSampleStyleSheet()
+    style = styles["Normal"]
+    style.fontName = "Courier"  # Use monospaced font like your original code
+    style.fontSize = 10
+    style.leading = 14  # line height
+
+    # Convert markdown-style bold (**text**) to HTML <b>text</b>
+    html_text = re.sub(r'\*\*(.+?)\*\*', r'<b>\1</b>', itinerary_text)
+
+    # Split the text into paragraphs by newlines
+    paragraphs = html_text.split('\n')
+
+    # Build the story (list of flowables)
+    story = []
+    for para in paragraphs:
+        if para.strip():
+            story.append(Paragraph(para, style))
+        else:
+            story.append(Paragraph(' ', style))
+        story.append(Spacer(1, 6))
+
+    doc.build(story)
+
+    return file_path
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
     # Header with gr.HTML for styles and animation