Update app.py

app.py

@@ -228,109 +228,182 @@ def create_pdf(proposal, output_path="proposal.pdf"):
 
 def create_slides(proposal):
     """Create PowerPoint slides from the proposal"""
-    prs = Presentation()
-    
-    # Extract title
-    title = extract_title(proposal)
-    
-    # Set up slide layouts
-    title_slide_layout = prs.slide_layouts[0]
-    section_title_layout = prs.slide_layouts[2]
-    content_layout = prs.slide_layouts[1]
-    
-    # Add title slide
-    title_slide = prs.slides.add_slide(title_slide_layout)
-    title_slide.shapes.title.text = f"Project Proposal: {title}"
-    subtitle = title_slide.placeholders[1]
-    subtitle.text = "Professional Project Proposal"
-    
-    # List of sections to look for
-    sections = [
-        "Executive Summary", 
-        "Project Background", 
-        "Goals and Objectives", 
-        "Methodology and Approach", 
-        "Timeline", 
-        "Budget Considerations", 
-        "Expected Outcomes", 
-        "Team and Resources", 
-        "Risk Assessment", 
-        "Conclusion"
-    ]
-    
-    # Extract sections using regex
-    section_matches = re.finditer(r'\*\*\d+\.\s+(.*?)\*\*\n\n(.*?)(?=\*\*\d+\.|\Z)', 
-                                 proposal, re.DOTALL)
-    
-    for match in section_matches:
-        section_title = match.group(1)
-        section_content = match.group(2).strip()
+    try:
+        prs = Presentation()
+        
+        # Extract title
+        title = extract_title(proposal)
         
-        # Add section title slide
-        section_slide = prs.slides.add_slide(section_title_layout)
-        section_slide.shapes.title.text = section_title
+        # Set up slide layouts
+        title_slide_layout = prs.slide_layouts[0]
+        section_title_layout = prs.slide_layouts[2]
+        content_layout = prs.slide_layouts[1]
         
-        # Split content into paragraphs
-        paragraphs = section_content.split('\n\n')
+        # Add title slide
+        title_slide = prs.slides.add_slide(title_slide_layout)
+        title_slide.shapes.title.text = f"Project Proposal: {title}"
+        subtitle = title_slide.placeholders[1]
+        subtitle.text = "Professional Project Proposal"
         
-        # Process each paragraph
-        current_slide = None
-        text_frame = None
-        paragraphs_on_slide = 0
+        # List of sections to look for
+        sections = [
+            "Executive Summary", 
+            "Project Background", 
+            "Goals and Objectives", 
+            "Methodology and Approach", 
+            "Timeline", 
+            "Budget Considerations", 
+            "Expected Outcomes", 
+            "Team and Resources", 
+            "Risk Assessment", 
+            "Conclusion"
+        ]
         
-        for para in paragraphs:
-            para = para.strip()
-            if not para:
-                continue
+        # Extract sections using regex
+        section_matches = re.finditer(r'\*\*\d+\.\s+(.*?)\*\*\n\n(.*?)(?=\*\*\d+\.|\Z)', 
+                                    proposal, re.DOTALL)
+        
+        for match in section_matches:
+            section_title = match.group(1)
+            section_content = match.group(2).strip()
             
-            # Start a new slide if needed
-            if current_slide is None or paragraphs_on_slide >= 3:
-                current_slide = prs.slides.add_slide(content_layout)
-                current_slide.shapes.title.text = section_title
-                text_frame = current_slide.placeholders[1].text_frame
-                paragraphs_on_slide = 0
+            # Add section title slide
+            section_slide = prs.slides.add_slide(section_title_layout)
+            section_slide.shapes.title.text = section_title
             
-            # Add the paragraph
-            p = text_frame.add_paragraph()
+            # Split content into paragraphs
+            paragraphs = section_content.split('\n\n')
             
-            # Check if it's a bullet point list
-            if re.match(r'^[\*\-]', para):
-                # Process bullet points
-                bullet_items = re.split(r'\n[\*\-]\s+', para)
-                for item in bullet_items:
-                    item = item.strip()
-                    if item:
-                        # Remove leading bullet if present
-                        item = re.sub(r'^[\*\-]\s+', '', item)
-                        bullet_p = text_frame.add_paragraph()
-                        bullet_p.text = item
-                        bullet_p.level = 1
-            else:
-                p.text = para
+            # Process each paragraph
+            current_slide = None
+            text_frame = None
+            paragraphs_on_slide = 0
             
-            paragraphs_on_slide += 1
+            for para in paragraphs:
+                para = para.strip()
+                if not para:
+                    continue
+                
+                # Start a new slide if needed
+                if current_slide is None or paragraphs_on_slide >= 3:
+                    current_slide = prs.slides.add_slide(content_layout)
+                    current_slide.shapes.title.text = section_title
+                    text_frame = current_slide.placeholders[1].text_frame
+                    paragraphs_on_slide = 0
+                
+                # Add the paragraph
+                p = text_frame.add_paragraph()
+                
+                # Check if it's a bullet point list
+                if re.match(r'^[\*\-]', para):
+                    # Process bullet points
+                    bullet_items = re.split(r'\n[\*\-]\s+', para)
+                    for item in bullet_items:
+                        item = item.strip()
+                        if item:
+                            # Remove leading bullet if present
+                            item = re.sub(r'^[\*\-]\s+', '', item)
+                            bullet_p = text_frame.add_paragraph()
+                            bullet_p.text = item
+                            bullet_p.level = 1
+                else:
+                    p.text = para
+                
+                paragraphs_on_slide += 1
+        
+        # Save the presentation
+        output_path = "proposal_slides.pptx"
+        prs.save(output_path)
+        return output_path
     
-    # Save the presentation
-    output_path = "proposal_slides.pptx"
-    prs.save(output_path)
-    return output_path
+    except Exception as e:
+        print(f"Error creating PowerPoint: {e}")
+        # Create a simple error presentation
+        try:
+            prs = Presentation()
+            slide = prs.slides.add_slide(prs.slide_layouts[0])
+            slide.shapes.title.text = "Error Creating Presentation"
+            subtitle = slide.placeholders[1]
+            subtitle.text = f"An error occurred: {str(e)}\nPlease try again with a different project description."
+            
+            output_path = "proposal_slides.pptx"
+            prs.save(output_path)
+        except:
+            # If even the error presentation fails, return a default path
+            pass
+        return output_path
 
 def process_input(description, project_type):
     """Process the input and generate both proposal, PDF and slides"""
-    # Check if input is too short
-    if len(description.strip()) < 10:
-        return "Please provide a more detailed project description (at least 10 characters).", None, None
-    
-    # Generate the proposal
-    proposal = generate_proposal(description, project_type)
-    
-    # Create the PDF
-    pdf_path = create_pdf(proposal)
-    
-    # Create the slides
-    ppt_path = create_slides(proposal)
-    
-    return proposal, pdf_path, ppt_path
+    try:
+        # Check if input is too short
+        if len(description.strip()) < 10:
+            return "Please provide a more detailed project description (at least 10 characters).", None, None
+        
+        # Map project type to specific type value
+        type_value = None
+        if project_type == "SaaS Performance Platform":
+            type_value = "saas_performance"
+        
+        # Generate the proposal
+        proposal = generate_proposal(description, type_value)
+        
+        # Create a basic PDF without any custom styles - fallback approach
+        try:
+            from reportlab.lib.pagesizes import letter
+            from reportlab.pdfgen import canvas
+            
+            # Basic PDF generation using canvas directly
+            c = canvas.Canvas("proposal.pdf", pagesize=letter)
+            width, height = letter
+            
+            # Title
+            c.setFont("Helvetica-Bold", 16)
+            c.drawCentredString(width/2, height-50, "Project Proposal")
+            
+            # Simple text rendering
+            c.setFont("Helvetica", 10)
+            y = height - 80
+            line_height = 14
+            
+            lines = proposal.split('\n')
+            for line in lines:
+                # Check for page break
+                if y < 50:
+                    c.showPage()
+                    y = height - 50
+                
+                # Check if it's a header (starts with **)
+                if line.startswith('**'):
+                    c.setFont("Helvetica-Bold", 12)
+                    # Remove ** markers
+                    line = line.replace('**', '')
+                else:
+                    c.setFont("Helvetica", 10)
+                
+                # Draw the line
+                if line.strip():
+                    c.drawString(50, y, line)
+                    y -= line_height
+            
+            c.save()
+            pdf_path = "proposal.pdf"
+        except Exception as pdf_error:
+            print(f"Error creating PDF: {pdf_error}")
+            pdf_path = None
+        
+        # Create the slides
+        try:
+            ppt_path = create_slides(proposal)
+        except Exception as ppt_error:
+            print(f"Error creating slides: {ppt_error}")
+            ppt_path = None
+        
+        return proposal, pdf_path, ppt_path
+    except Exception as e:
+        error_message = f"An error occurred: {str(e)}\nPlease try again with a different project description."
+        print(f"Error in process_input: {e}")
+        return error_message, None, None
 
 # Create Gradio interface
 def create_interface():