Update app.py

app.py

@@ -473,18 +473,21 @@ def generate_pdf(documentation):
     pdf = FPDF()
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
-    pdf.set_font("Arial", size=12)
+    pdf.set_font("Courier", size=12)  # Monospaced font for IDE-like appearance
 
-    # Add headers and content
+    # Process and format content
     for line in documentation.splitlines():
+        # Remove asterisks and backticks
+        line = line.replace("*", "").replace("`", "'")
+
+        # Detect headers and adjust formatting
         if line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
            line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
-            pdf.set_font("Arial", style="B", size=14)
-        elif line.startswith("*") or line.startswith("- **Function**"):
-            pdf.set_font("Arial", style="B", size=12)
+            pdf.set_font("Courier", style="B", size=14)
+            pdf.cell(0, 10, line, ln=True)
+            pdf.set_font("Courier", size=12)  # Reset font for content
         else:
-            pdf.set_font("Arial", size=12)
-        pdf.multi_cell(0, 10, line)
+            pdf.multi_cell(0, 10, line.strip())  # Handle regular content
 
     # Save and download the PDF
     pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
@@ -499,15 +502,33 @@ def generate_pdf(documentation):
 
 # Helper function to generate Markdown file
 def generate_markdown_file(documentation):
-    # Format the documentation
-    formatted_documentation = documentation.replace("**", "**").replace(":", ":\n")
+    """
+    Generate a Markdown file from the documentation with IDE-like formatting.
+    Adjustments:
+    - Removes asterisks (*)
+    - Replaces backticks (`) with single quotes (')
+    - Removes extra spaces between section headers and their content
+    - Removes Markdown-specific delimiters (e.g., ``` at the top and bottom)
+    """
+    # Process and format content
+    formatted_documentation = ""
+    for line in documentation.splitlines():
+        # Remove asterisks and backticks
+        line = line.replace("*", "").replace("`", "'")
+
+        # Add a newline only for headers for better structure in Markdown
+        if line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
+           line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
+            formatted_documentation += f"## {line.strip()}\n"
+        else:
+            formatted_documentation += f"{line.strip()}\n"
 
-    # Save as a temporary Markdown file
+    # Save to a temporary Markdown file
     markdown_file = tempfile.NamedTemporaryFile(delete=False, suffix=".md")
     with open(markdown_file.name, "w") as f:
         f.write(formatted_documentation)
 
-    # Download the Markdown file
+    # Allow the user to download the Markdown file
     st.download_button(
         label="Download Markdown File",
         data=open(markdown_file.name, "rb").read(),