Update app.py

app.py

@@ -77,6 +77,16 @@ def generate_notes(text):
     )
     return response.choices[0].message.content.strip()
 
+# Function to generate additional sections
+def generate_section(title, text):
+    prompt = f"Generate a section titled '{title}' with 3-6 sentences based on the following text:\n\n{text}"
+    response = openai.ChatCompletion.create(
+        model='gpt-3.5-turbo',
+        messages=[{'role': 'user', 'content': prompt}],
+        max_tokens=500,
+    )
+    return response.choices[0].message.content.strip()
+
 # Main app
 def main():
     st.set_page_config(layout="wide")
@@ -132,18 +142,34 @@ def main():
                     f"<p>{summary}</p>" +
                     "</div>", unsafe_allow_html=True)
 
-        st.markdown("### Learning Outcomes:")
-        st.markdown("<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
-                    "<ul>" +
-                    "<li>Understand key points discussed in the lecture.</li>" +
-                    "<li>Be able to explain concepts with clarity.</li>" +
-                    "<li>Implement ideas in practical scenarios.</li>" +
-                    "</ul>" +
+        st.markdown("### Key Concepts:")
+        key_concepts = generate_section("Key Concepts", summary)
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
+                    f"<p>{key_concepts}</p>" +
                     "</div>", unsafe_allow_html=True)
 
-        st.markdown("### Lecture Slides:")
-        st.markdown("<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
-                    "<p>Click here to upload or access slides.</p>" +
+        st.markdown("### Key Takeaways:")
+        key_takeaways = generate_section("Key Takeaways", summary)
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
+                    f"<p>{key_takeaways}</p>" +
+                    "</div>", unsafe_allow_html=True)
+
+        st.markdown("### Case Studies/Examples:")
+        case_studies = generate_section("Case Studies/Examples", summary)
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
+                    f"<p>{case_studies}</p>" +
+                    "</div>", unsafe_allow_html=True)
+
+        st.markdown("### Glossary:")
+        glossary = generate_section("Glossary", summary)
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
+                    f"<p>{glossary}</p>" +
+                    "</div>", unsafe_allow_html=True)
+
+        st.markdown("### FAQs:")
+        faqs = generate_section("FAQs", summary)
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
+                    f"<p>{faqs}</p>" +
                     "</div>", unsafe_allow_html=True)
 
 if __name__ == "__main__":