Update app.py

app.py

@@ -356,7 +356,8 @@ def generate_detailed_documentation(file_contents, functionality_description):
         '
         Functionality Flow:
         <Explain the sequence of functions and data flow. If there are multiple initial input points required for the user-defined functionality, show the flow of each one.
-        An example flow description is: 1. User enters 3. 2. foo(3) is called. 3. foo(3) calls bar() and fubar(3) to return a list filled with... 7. dubar(97) returns 'yes' to the user>
+        An example flow description is: 1. User enters 3. 2. foo(3) is called. 3. foo(3) calls bar() and fubar(3) to return a list filled with... 7. dubar(97) returns 'yes' to the user
+        make sure there is no empty line between 'Functionality Flow:' and what is to be placed in this section>
         '
     4. Generate detailed documentation for each function in the codebase:
         '
@@ -385,11 +386,12 @@ def generate_detailed_documentation(file_contents, functionality_description):
 def process_gemini_output(output):
     """
     Processes the raw output from Gemini to format it appropriately.
-    
+
     - Removes ``` at the top and bottom of the output.
     - Removes all pairs of **.
     - Replaces ` used as a quote with '.
     - Replaces leading '*' with '-' for lists.
+    - Makes function names under "Function Documentation" bold.
     
     Args:
         output (str): The raw Gemini output.
@@ -412,14 +414,19 @@ def process_gemini_output(output):
     processed_lines = []
     for line in lines:
         stripped_line = line.strip()
+        
+        # Replace '*' at the start of lines with '-'
         if stripped_line.startswith("*"):
-            # Only replace if the '*' is at the start or followed by whitespace
             processed_lines.append(line.replace("*", "-", 1))
+        
+        # Make function names bold (lines starting with "- '" and ending with "':")
+        elif stripped_line.startswith("- '") and stripped_line.endswith("':"):
+            processed_lines.append(f"**{line.strip()}**")
         else:
             processed_lines.append(line)
     
     return "\n".join(processed_lines)
-    
+
 def generate_documentation_page():
     st.subheader(f"Generate Documentation for {st.session_state.current_project}")
     st.write("Enter the functionality or parts of the project for which you'd like to generate documentation.")
@@ -507,21 +514,32 @@ def generate_documentation_page():
 
 # Helper function to generate PDF
 def generate_pdf(documentation):
+    """
+    Generates a PDF from the processed documentation with a programming IDE font (Courier).
+
+    - Function names are bolded.
+    - Uses Courier for all text.
+
+    Args:
+        documentation (str): The formatted 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)  # Use Courier for IDE-like font
 
     # Add headers and content
     for line in documentation.splitlines():
         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.multi_cell(0, 10, line)
+        elif line.startswith("**- '") and line.endswith("':**"):  # Bold function names
+            pdf.set_font("Courier", style="B", size=12)
+            pdf.multi_cell(0, 10, line.replace("**", ""))
         else:
-            pdf.set_font("Arial", size=12)
-        pdf.multi_cell(0, 10, line)
+            pdf.set_font("Courier", size=12)
+            pdf.multi_cell(0, 10, line)
 
     # Save and download the PDF
     pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")