Update app.py

app.py

@@ -386,12 +386,11 @@ 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.
@@ -414,19 +413,14 @@ 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.")
@@ -514,32 +508,21 @@ 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("Courier", size=12)  # Use Courier for IDE-like font
+    pdf.set_font("Arial", size=12)
 
     # 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("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("**", ""))
+            pdf.set_font("Arial", style="B", size=14)
+        elif line.startswith("*") or line.startswith("- **Function**"):
+            pdf.set_font("Arial", style="B", size=12)
         else:
-            pdf.set_font("Courier", size=12)
-            pdf.multi_cell(0, 10, line)
+            pdf.set_font("Arial", size=12)
+        pdf.multi_cell(0, 10, line)
 
     # Save and download the PDF
     pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
@@ -634,4 +617,3 @@ def project_view_page():
 if __name__ == "__main__":
     main()
 
-