Update app.py

app.py

@@ -7,7 +7,9 @@ from fpdf import FPDF
 import pandas as pd
 import re
 from io import BytesIO
-import base64
+from pptx import Presentation
+from pptx.util import Inches, Pt
+from pptx.enum.text import PP_ALIGN
 
 # Load Features from CSV (curate a subset for demo clarity)
 features_df = pd.read_csv("Feature-Description.csv")
@@ -184,38 +186,113 @@ def generate_documentation(code_text, language, export_format, selected_features
         summary_ids = mdl.generate(inputs, max_length=64, num_beams=2, early_stopping=True)
         summary = tok.decode(summary_ids[0], skip_special_tokens=True)
     
-    extra_sections = ""
+    extra_sections = {}
     for feature in selected_features:
         if feature in feature_functions:
             if feature == "UML Diagram Generation":
-                extra_sections += f"\n**{feature}:**\n{feature_functions[feature](code_text, language)}\n"
+                extra_sections[feature] = feature_functions[feature](code_text, language)
             else:
-                extra_sections += f"\n**{feature}:**\n{feature_functions[feature](code_text)}\n"
+                extra_sections[feature] = feature_functions[feature](code_text)
     
+    # Markdown output
     doc_output = f"""### AI-Generated Documentation
 
 {summary}
 
 **Code Complexity Score:** {complexity_score:.2f} (0=low, 1=high)
 
-{extra_sections}
 """
+    for feature, content in extra_sections.items():
+        doc_output += f"\n**{feature}:**\n{content}\n"
     
     if export_format == "Markdown":
-        return doc_output
+        return doc_output, None, None
     elif export_format == "PDF":
         pdf_filename = "/tmp/generated_doc.pdf"
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", size=10)
-        for line in doc_output.split('\n'):
-            # Handle special characters
-            safe_line = line.encode('latin-1', 'replace').decode('latin-1')
-            pdf.multi_cell(0, 6, txt=safe_line)
+        
+        # Title
+        pdf.set_font("Arial", 'B', 16)
+        pdf.cell(0, 10, txt="AI-Generated Documentation", ln=True, align='C')
+        pdf.ln(5)
+        
+        # Summary
+        pdf.set_font("Arial", 'B', 12)
+        pdf.cell(0, 8, txt="Summary:", ln=True)
+        pdf.set_font("Arial", size=10)
+        pdf.multi_cell(0, 6, txt=summary)
+        pdf.ln(3)
+        
+        # Complexity Score
+        pdf.set_font("Arial", 'B', 11)
+        pdf.cell(0, 8, txt=f"Code Complexity Score: {complexity_score:.2f}", ln=True)
+        pdf.ln(3)
+        
+        # Extra sections
+        for feature, content in extra_sections.items():
+            pdf.set_font("Arial", 'B', 11)
+            pdf.cell(0, 8, txt=feature + ":", ln=True)
+            pdf.set_font("Arial", size=9)
+            safe_content = content.replace('```mermaid', '').replace('```', '').encode('latin-1', 'replace').decode('latin-1')
+            pdf.multi_cell(0, 5, txt=safe_content)
+            pdf.ln(2)
+        
         pdf.output(pdf_filename)
-        return pdf_filename
+        return None, pdf_filename, None
+    
+    elif export_format == "PPT":
+        ppt_filename = "/tmp/generated_doc.pptx"
+        prs = Presentation()
+        prs.slide_width = Inches(10)
+        prs.slide_height = Inches(7.5)
+        
+        # Slide 1: Title Slide
+        title_slide_layout = prs.slide_layouts[0]
+        slide = prs.slides.add_slide(title_slide_layout)
+        title = slide.shapes.title
+        subtitle = slide.placeholders[1]
+        title.text = "AI-Generated Code Documentation"
+        subtitle.text = f"Language: {language}\nComplexity Score: {complexity_score:.2f}"
+        
+        # Slide 2: Summary
+        bullet_slide_layout = prs.slide_layouts[1]
+        slide = prs.slides.add_slide(bullet_slide_layout)
+        shapes = slide.shapes
+        title_shape = shapes.title
+        body_shape = shapes.placeholders[1]
+        title_shape.text = "Code Summary"
+        tf = body_shape.text_frame
+        tf.text = summary
+        
+        # Slide 3+: Feature sections
+        for feature, content in extra_sections.items():
+            slide = prs.slides.add_slide(bullet_slide_layout)
+            shapes = slide.shapes
+            title_shape = shapes.title
+            body_shape = shapes.placeholders[1]
+            title_shape.text = feature
+            
+            tf = body_shape.text_frame
+            clean_content = content.replace('```mermaid', '').replace('```', '')
+            
+            # Split content into lines and add as bullets
+            lines = clean_content.split('\n')
+            for i, line in enumerate(lines[:10]):  # Limit to 10 lines per slide
+                if i == 0:
+                    tf.text = line.strip()
+                else:
+                    if line.strip():
+                        p = tf.add_paragraph()
+                        p.text = line.strip()
+                        p.level = 0
+        
+        prs.save(ppt_filename)
+        return None, None, ppt_filename
+    
     else:
-        return doc_output
+        return doc_output, None, None
 
 
 def process_uploaded_file(uploaded_file, language, export_format, selected_features):
@@ -356,7 +433,7 @@ with gr.Blocks(css=custom_css, elem_id="container") as demo:
     file_input = gr.File(label="Upload Code File (.py, .js, .java)", file_types=[".py", ".js", ".java"])
     code_input = gr.Textbox(label="Or Paste Code Here", lines=8, max_lines=15, placeholder="Paste your code snippet here...")
     language_dropdown = gr.Dropdown(label="Select Language", choices=["Python", "JavaScript", "Java", "Other"], value="Python")
-    export_dropdown = gr.Dropdown(label="Export Format", choices=["Markdown", "PDF"], value="Markdown")
+    export_dropdown = gr.Dropdown(label="Export Format", choices=["Markdown", "PDF", "PPT"], value="Markdown")
     feature_options = gr.CheckboxGroup(
         label="Select Features to Include",
         choices=[f["Feature"] for f in features_list],
@@ -367,27 +444,34 @@ with gr.Blocks(css=custom_css, elem_id="container") as demo:
     )
     generate_btn = gr.Button("Generate Documentation", elem_id="generate-btn")
     output_box = gr.Textbox(label="Generated Documentation", lines=10, max_lines=20, interactive=False, show_copy_button=True)
-    pdf_output = gr.File(label="Download PDF", visible=False)
+    
+    with gr.Row():
+        pdf_output = gr.File(label="Download PDF", visible=False)
+        ppt_output = gr.File(label="Download PPT", visible=False)
+    
     gr.HTML("<div id='credits'>Credits: Sreelekha Putta</div>")
 
 
     def on_generate(file_obj, code_str, language, export_format, selected_features):
         if file_obj is not None:
-            result = process_uploaded_file(file_obj, language, export_format, selected_features)
+            markdown_result, pdf_result, ppt_result = process_uploaded_file(file_obj, language, export_format, selected_features)
         elif code_str.strip() != "":
-            result = generate_documentation(code_str, language, export_format, selected_features)
+            markdown_result, pdf_result, ppt_result = generate_documentation(code_str, language, export_format, selected_features)
         else:
-            return "Please upload a file or paste code to generate documentation.", None
+            return "Please upload a file or paste code to generate documentation.", gr.update(visible=False), gr.update(visible=False)
+        
         if export_format == "PDF":
-            return None, gr.update(value=result, visible=True)
+            return None, gr.update(value=pdf_result, visible=True), gr.update(visible=False)
+        elif export_format == "PPT":
+            return None, gr.update(visible=False), gr.update(value=ppt_result, visible=True)
         else:
-            return result, gr.update(visible=False)
+            return markdown_result, gr.update(visible=False), gr.update(visible=False)
 
 
     generate_btn.click(
         on_generate,
         inputs=[file_input, code_input, language_dropdown, export_dropdown, feature_options],
-        outputs=[output_box, pdf_output]
+        outputs=[output_box, pdf_output, ppt_output]
     )