Update src/streamlit_app.py

src/streamlit_app.py

@@ -23,6 +23,10 @@ if 'presentation_title' not in st.session_state:
     st.session_state.presentation_title = ""
 if 'openai_api_key' not in st.session_state:
     st.session_state.openai_api_key = ""
+if 'style_instructions' not in st.session_state:
+    st.session_state.style_instructions = ""
+if 'content_instructions' not in st.session_state:
+    st.session_state.content_instructions = ""
 if 'css_style' not in st.session_state:
     st.session_state.css_style = """
     body {
@@ -81,11 +85,35 @@ if 'css_style' not in st.session_state:
 # Function to create presentation plan using LLM as "Planning Agent"
 def create_presentation_plan(topic, num_slides, key_points):
     try:
-        # Simulating LLM call - In production, you would replace this with an actual LLM API call
-        # For example, with OpenAI's API or another LLM provider
-        
-        # For demo, simulate an LLM response with a predefined structure
-        time.sleep(1)  # Simulate API call time
+        if st.session_state.openai_api_key:
+            # Use actual OpenAI API
+            client = openai.OpenAI(api_key=st.session_state.openai_api_key)
+            response = client.chat.completions.create(
+                model="gpt-3.5-turbo",  # You can change to gpt-4 if available
+                messages=[
+                    {"role": "system", "content": "You are a presentation planning expert. Create a detailed presentation outline."},
+                    {"role": "user", "content": f"Create a presentation plan on '{topic}' with {num_slides} slides. Key points to include: {key_points}. Return your response as a JSON object with this structure: {{\"title\": \"Presentation Title\", \"num_slides\": number, \"slides\": [{{\"slide_number\": 1, \"title\": \"Slide Title\", \"type\": \"title_slide\", \"content\": \"Content description\"}}]}}"}
+                ],
+                response_format={"type": "json_object"}
+            )
+            # Parse the JSON response
+            try:
+                return json.loads(response.choices[0].message.content)
+            except json.JSONDecodeError:
+                # Fallback to simulated response if JSON parsing fails
+                st.warning("Could not parse LLM response as JSON. Using simulated plan instead.")
+                return simulate_presentation_plan(topic, num_slides, key_points)
+        else:
+            # Fallback to simulated response
+            return simulate_presentation_plan(topic, num_slides, key_points)
+    except Exception as e:
+        st.error(f"Error creating presentation plan: {str(e)}")
+        # Fallback to simulated response
+        return simulate_presentation_plan(topic, num_slides, key_points)
+
+def simulate_presentation_plan(topic, num_slides, key_points):
+    # Simulate an LLM response with a predefined structure
+    time.sleep(1)  # Simulate API call time
         
         slides = []
         # Always create a title slide
@@ -399,11 +427,25 @@ with tabs[3]:
         st.markdown(get_slide_download_link(), unsafe_allow_html=True)
         st.info("This will download a ZIP file containing all your slides as HTML files with consistent styling. You can open each slide in a browser or use them in your presentation software.")
 
-# Sidebar with instructions
-st.sidebar.title("How It Works")
-st.sidebar.markdown("""
-### AI PowerPoint Maker
+# Sidebar with instructions and settings
+st.sidebar.title("AI PowerPoint Maker")
+
+# OpenAI API Key input
+st.sidebar.subheader("OpenAI API Settings")
+api_key = st.sidebar.text_input("Enter your OpenAI API Key", 
+                               value=st.session_state.openai_api_key,
+                               type="password", 
+                               help="Your API key will be used to generate better presentation content and slides.")
+
+if api_key:
+    st.session_state.openai_api_key = api_key
+    st.sidebar.success("✅ API Key set successfully!")
+else:
+    st.sidebar.info("ℹ️ No API Key provided. The app will use simulated responses.")
 
+st.sidebar.markdown("---")
+st.sidebar.subheader("How It Works")
+st.sidebar.markdown("""
 This app uses a multi-agent approach to create presentations:
 
 1. **Planning Agent**: Develops the structure and content flow of your presentation
@@ -424,4 +466,4 @@ This app uses a multi-agent approach to create presentations:
 """)
 
 st.sidebar.markdown("---")
-st.sidebar.info("Note: In a production version, this would use actual API calls to an LLM service for both the planning and HTML creation steps. This demo uses simulated responses for demonstration purposes.")
+st.sidebar.info("When the OpenAI API key is provided, this app will use the actual OpenAI API for both the planning and HTML creation steps. Without an API key, it will use simulated responses for demonstration purposes.")