Update app.py

app.py

@@ -130,7 +130,7 @@ def generate_notes(transcript):
     pdf_path = create_pdf(res)
     return pdf_path
 
-def create_pdf(transcript, long_questions, short_questions, mcqs):
+def create_pdf(question):
     pdf = FPDF()
     pdf.add_page()
     
@@ -140,31 +140,31 @@ def create_pdf(transcript, long_questions, short_questions, mcqs):
 
     # Add transcription content
     pdf.set_font("Arial", "", 12)
-    pdf.multi_cell(0, 10, f"Transcription:\n{transcript.encode('latin1', 'replace').decode('latin1')}\n\n")
-
-    # Add long questions
-    pdf.set_font("Arial", "B", 14)
-    pdf.cell(200, 10, "Long Questions", ln=True)
-    pdf.set_font("Arial", "", 12)
-    for question in long_questions:
-        pdf.multi_cell(0, 10, f"- {question.encode('latin1', 'replace').decode('latin1')}\n")
-
-    # Add short questions
-    pdf.set_font("Arial", "B", 14)
-    pdf.cell(200, 10, "Short Questions", ln=True)
-    pdf.set_font("Arial", "", 12)
-    for question in short_questions:
-        pdf.multi_cell(0, 10, f"- {question.encode('latin1', 'replace').decode('latin1')}\n")
-
-    # Add MCQs
-    pdf.set_font("Arial", "B", 14)
-    pdf.cell(200, 10, "Multiple Choice Questions (MCQs)", ln=True)
-    pdf.set_font("Arial", "", 12)
-    for mcq in mcqs:
-        pdf.multi_cell(0, 10, f"Q: {mcq['question'].encode('latin1', 'replace').decode('latin1')}")
-        for option in mcq["options"]:
-            pdf.multi_cell(0, 10, f"   - {option.encode('latin1', 'replace').decode('latin1')}")
-        pdf.multi_cell(0, 10, f"Answer: {mcq['answer'].encode('latin1', 'replace').decode('latin1')}\n")
+    pdf.multi_cell(0, 10, f"Transcription:\n{question.encode('latin1', 'replace').decode('latin1')}\n\n")
+
+    # # Add long questions
+    # pdf.set_font("Arial", "B", 14)
+    # pdf.cell(200, 10, "Long Questions", ln=True)
+    # pdf.set_font("Arial", "", 12)
+    # for question in long_questions:
+    #     pdf.multi_cell(0, 10, f"- {question.encode('latin1', 'replace').decode('latin1')}\n")
+
+    # # Add short questions
+    # pdf.set_font("Arial", "B", 14)
+    # pdf.cell(200, 10, "Short Questions", ln=True)
+    # pdf.set_font("Arial", "", 12)
+    # for question in short_questions:
+    #     pdf.multi_cell(0, 10, f"- {question.encode('latin1', 'replace').decode('latin1')}\n")
+
+    # # Add MCQs
+    # pdf.set_font("Arial", "B", 14)
+    # pdf.cell(200, 10, "Multiple Choice Questions (MCQs)", ln=True)
+    # pdf.set_font("Arial", "", 12)
+    # for mcq in mcqs:
+    #     pdf.multi_cell(0, 10, f"Q: {mcq['question'].encode('latin1', 'replace').decode('latin1')}")
+    #     for option in mcq["options"]:
+    #         pdf.multi_cell(0, 10, f"   - {option.encode('latin1', 'replace').decode('latin1')}")
+    #     pdf.multi_cell(0, 10, f"Answer: {mcq['answer'].encode('latin1', 'replace').decode('latin1')}\n")
 
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
         pdf.output(temp_pdf.name)