Update app.py

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
 from pptx import Presentation
+from collections import defaultdict
 
 # Function to create PowerPoint presentation based on user inputs
-def create_presentation(sections):
+def create_presentation(title_slide_info, sections):
     prs = Presentation()
 
     # Add a title slide
@@ -10,11 +11,11 @@ def create_presentation(sections):
     title_slide = prs.slides.add_slide(title_slide_layout)
     title = title_slide.shapes.title
     subtitle = title_slide.placeholders[1]
-    title.text = "Your Presentation Title"
-    subtitle.text = "Your Name\nYour Institution\nDate"
+    title.text = title_slide_info['title']
+    subtitle.text = f"{title_slide_info['subtitle']}\n{title_slide_info['institution']}\n{title_slide_info['date']}"
 
     # Add content slides based on sections
-    for section in sections:
+    for section in sections.values():
         slide_layout = prs.slide_layouts[1]
         slide = prs.slides.add_slide(slide_layout)
         title_placeholder = slide.shapes.title
@@ -31,22 +32,19 @@ def create_presentation(sections):
     return filename
 
 # Streamlit UI
-st.title('Custom Presentation Generator')
+st.sidebar.title('Presentation Settings')
+presentation_type = st.sidebar.selectbox('Select Presentation Type', ['Research Presentation', 'Proposal Defense'])
 
-# Select template type
-template_type = st.selectbox('Select Presentation Type', ['Research Presentation', 'Proposal Defense'])
-
-# Predefined sections based on template type
-if template_type == 'Research Presentation':
-    default_sections = [
+# Template selection
+templates = {
+    'Research Presentation': [
         {"title": "Introduction", "content": ["Background", "Problem Statement", "Purpose"]},
         {"title": "Literature Review", "content": ["Previous Studies", "Key Theories", "Gaps in Research"]},
         {"title": "Methodology", "content": ["Research Design", "Data Collection", "Data Analysis"]},
         {"title": "Results", "content": ["Data Presentation", "Statistical Analysis", "Interpretation"]},
         {"title": "Discussion and Conclusion", "content": ["Summary of Findings", "Implications", "Recommendations for Future Research"]}
-    ]
-elif template_type == 'Proposal Defense':
-    default_sections = [
+    ],
+    'Proposal Defense': [
         {"title": "Introduction", "content": ["Background Context", "Problem Statement", "Research Questions", "Purpose and Significance"]},
         {"title": "Literature Review", "content": ["Key Theories and Concepts", "Current State of Research", "Gaps in Literature"]},
         {"title": "Research Design and Methodology", "content": ["Research Framework", "Data Collection Methods", "Analysis Plan", "Ethical Considerations"]},
@@ -56,24 +54,53 @@ elif template_type == 'Proposal Defense':
         {"title": "Conclusion", "content": ["Summary of Key Points", "Reiteration of Research Importance", "Next Steps"]},
         {"title": "Questions and Discussion", "content": ["Anticipated Questions", "Potential Answers", "Further Clarifications"]}
     ]
+}
+
+default_sections = templates[presentation_type]
+
+# Title slide configuration
+st.sidebar.header("Title Slide")
+title = st.sidebar.text_input("Title", "Your Presentation Title")
+subtitle = st.sidebar.text_input("Subtitle", "Your Name")
+institution = st.sidebar.text_input("Institution", "Your Institution")
+presentation_date = st.sidebar.text_input("Date", "Presentation Date")
+title_slide_info = {'title': title, 'subtitle': subtitle, 'institution': institution, 'date': presentation_date}
+
+# Sections configuration
+st.title('Customize Your Presentation')
+sections = defaultdict(lambda: {'title': '', 'contents': []})  # Uses defaultdict to handle dynamic section addition
+section_keys = st.session_state.get('section_keys', list(default_sections.keys()))
+new_section_key = st.text_input("Add New Section Title", "")
+
+if new_section_key:
+    section_keys.append(new_section_key)
+    st.session_state['section_keys'] = section_keys  # Update session state
+
+for i, key in enumerate(section_keys):
+    if key in default_sections:
+        default_section = default_sections[key]
+        section_title = default_section['title']
+        default_content = default_section['content']
+    else:
+        section_title, default_content = key, [""]
 
-sections = []
-for i, default_section in enumerate(default_sections):
-    st.markdown(f"### {default_section['title']}")
-    section_title = st.text_input(f"Title", value=default_section['title'], key=f'title_{i}')
+    st.markdown(f"### {section_title}")
+    section_title = st.text_input(f"Title", value=section_title, key=f'title_{i}')
     contents = []
-    for j, content in enumerate(default_section['content']):
-        bullet_point = st.text_area(f"Bullet Point {j+1}", value=content, key=f'section_{i}_bullet_{j}', height=75)
-        contents.append(bullet_point)
-    sections.append({"title": section_title, "contents": contents})
+    for j, content in enumerate(default_content + [""]):  # Extra empty string for adding new bullet points
+        bullet_point = st.text_area(f"Bullet Point {j+1}", value=content, height=75, key=f'section_{i}_bullet_{j}')
+        if bullet_point:  # Only add bullet points with content
+            contents.append(bullet_point)
+    sections[i]['title'] = section_title
+    sections[i]['contents'] = contents
 
 # Generate presentation
 if st.button('Generate Presentation'):
-    file_path = create_presentation(sections)
+    file_path = create_presentation(title_slide_info, sections)
     with open(file_path, "rb") as file:
-        btn = st.download_button(
+        st.download_button(
              label="Download Presentation",
              data=file,
-             file_name='your_presentation.pptx',
+             file_name='custom_presentation.pptx',
              mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
          )