Vaishnav14220 commited on
Commit
a49be2e
·
1 Parent(s): d6f2050

Fix PDF download by returning file path instead of BytesIO object

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -5,6 +5,8 @@ from fuzzywuzzy import process
5
  from rdkit import Chem
6
  from rdkit.Chem import AllChem, Draw
7
  import io
 
 
8
  from reportlab.pdfgen import canvas
9
  from reportlab.lib.pagesizes import letter
10
  from reportlab.lib.styles import getSampleStyleSheet
@@ -48,8 +50,11 @@ def generate_reaction_svg(name):
48
  return "Reaction not found."
49
 
50
  def generate_all_reactions_pdf():
51
- pdf_output = io.BytesIO()
52
- doc = SimpleDocTemplate(pdf_output, pagesize=letter)
 
 
 
53
  styles = getSampleStyleSheet()
54
  story = []
55
 
@@ -84,8 +89,7 @@ def generate_all_reactions_pdf():
84
  story.append(Spacer(1, 12))
85
 
86
  doc.build(story)
87
- pdf_output.seek(0)
88
- return pdf_output
89
 
90
  def search_by_reaction_name(query):
91
  if not query:
 
5
  from rdkit import Chem
6
  from rdkit.Chem import AllChem, Draw
7
  import io
8
+ import tempfile
9
+ import os
10
  from reportlab.pdfgen import canvas
11
  from reportlab.lib.pagesizes import letter
12
  from reportlab.lib.styles import getSampleStyleSheet
 
50
  return "Reaction not found."
51
 
52
  def generate_all_reactions_pdf():
53
+ # Create temporary file
54
+ temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
55
+ temp_file.close()
56
+
57
+ doc = SimpleDocTemplate(temp_file.name, pagesize=letter)
58
  styles = getSampleStyleSheet()
59
  story = []
60
 
 
89
  story.append(Spacer(1, 12))
90
 
91
  doc.build(story)
92
+ return temp_file.name
 
93
 
94
  def search_by_reaction_name(query):
95
  if not query: