Update app.py

app.py

@@ -132,15 +132,55 @@ def process_pdfs_and_generate_question_paper(uploaded_files, name, number_of_sec
 
     return response
 
-# Function to save the generated question paper to a Word document
-def save_to_word(text, file_name):
+# # Function to save the generated question paper to a Word document
+# def save_to_word(text, file_name):
+#     doc = Document()
+#     doc.add_heading("Generated Question Paper", 0)
+#     lines = text.split('\n')
+#     for line in lines:
+#         doc.add_paragraph(line)
+#     doc.save(file_name)
+#     return file_name
+
+def save_markdown_to_word(markdown_text, file_name):
+    # Create a new Document
     doc = Document()
-    doc.add_heading("Generated Question Paper", 0)
-    lines = text.split('\n')
+
+    # Add a title for the markdown content
+    doc.add_heading(f"{name}", 0)
+
+    # Replace bullet points and indent text
+    text = markdown_text.replace('•', '  *')
+    indented_text = textwrap.indent(text, '', predicate=lambda _: True)
+
+    # Split the text into lines
+    lines = indented_text.split('\n')
+
     for line in lines:
-        doc.add_paragraph(line)
+        # Check if the line contains text surrounded by ** for bold formatting
+        if '**' in line:
+            # Remove the ** and make the text bold
+            bold_text = re.sub(r'\*\*(.*?)\*\*', r'\1', line)  # Remove the ** and keep the text inside
+            # Add the line with bold formatting
+            p = doc.add_paragraph()
+            run = p.add_run(bold_text)
+            run.bold = True
+
+        elif '*' in line:
+            # Remove the * and make the text italic
+            italic_text = re.sub(r'\*(.*?)\*', r'\1', line)  # Remove the * and keep the text inside
+            # Add the line with italic formatting
+            p = doc.add_paragraph()
+            run = p.add_run(italic_text)
+            run.italic = True
+
+        else:
+            # Add regular lines
+            doc.add_paragraph(line)
+
+    # Save the document
     doc.save(file_name)
-    return file_name
+    print(f"Markdown content saved to {file_name}")
 
 def main_interface(uploaded_files, name, number_of_sections, difficulty_level, *args):
     sections = []
@@ -155,7 +195,7 @@ def main_interface(uploaded_files, name, number_of_sections, difficulty_level, *
     calculated_marks = sum(section['num_questions'] * section['marks_per_question'] for section in sections)
 
     response = process_pdfs_and_generate_question_paper(uploaded_files, name, number_of_sections, sections, difficulty_level, calculated_marks)
-    word_file = save_to_word(response.text, f"{name}.docx")
+    word_file = save_markdown_to_word(response.text, f"{name}.docx")
     return response.text, word_file