ikraamkb commited on
Commit
3cd8ec3
·
verified ·
1 Parent(s): a418dd2

Update Summarization/app.py

Browse files
Files changed (1) hide show
  1. Summarization/app.py +12 -6
Summarization/app.py CHANGED
@@ -279,8 +279,14 @@ def text_to_speech(text: str):
279
  except Exception:
280
  return ""
281
 
282
- def create_pdf(summary: str, filename: str):
283
  try:
 
 
 
 
 
 
284
  pdf = FPDF()
285
  pdf.add_page()
286
  pdf.set_font("Arial", 'B', 16)
@@ -290,13 +296,13 @@ def create_pdf(summary: str, filename: str):
290
  pdf.ln(10)
291
  pdf.set_font("Arial", size=10)
292
  pdf.multi_cell(0, 10, txt=summary)
293
- temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
294
- pdf.output(temp_pdf.name)
295
- return temp_pdf.name
296
- except Exception:
 
297
  return ""
298
 
299
- # --- Public API Function ---
300
 
301
  async def summarize_document(file: UploadFile, length: str = "medium"):
302
  try:
 
279
  except Exception:
280
  return ""
281
 
282
+ def create_pdf(summary: str, filename: str) -> str:
283
  try:
284
+ safe_dir = os.path.join(tempfile.gettempdir(), "aidan_files")
285
+ os.makedirs(safe_dir, exist_ok=True)
286
+
287
+ safe_filename = f"{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{filename}.pdf"
288
+ safe_path = os.path.join(safe_dir, safe_filename)
289
+
290
  pdf = FPDF()
291
  pdf.add_page()
292
  pdf.set_font("Arial", 'B', 16)
 
296
  pdf.ln(10)
297
  pdf.set_font("Arial", size=10)
298
  pdf.multi_cell(0, 10, txt=summary)
299
+
300
+ pdf.output(safe_path)
301
+ return safe_path
302
+ except Exception as e:
303
+ print(f"[PDF ERROR] {e}")
304
  return ""
305
 
 
306
 
307
  async def summarize_document(file: UploadFile, length: str = "medium"):
308
  try: