Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,148 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from utils import run_agent_sync
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+st.set_page_config(page_title="MCP POC", page_icon="🤖", layout="wide")
+
+st.title("Model Context Protocol(MCP) - Learning Path Generator")
+
+# Initialize session state for progress
+if 'current_step' not in st.session_state:
+    st.session_state.current_step = ""
+if 'progress' not in st.session_state:
+    st.session_state.progress = 0
+if 'last_section' not in st.session_state:
+    st.session_state.last_section = ""
+if 'is_generating' not in st.session_state:
+    st.session_state.is_generating = False
+
+# Sidebar for API and URL configuration
+st.sidebar.header("Configuration")
+
+# API Key input
+google_api_key = st.sidebar.text_input("Google API Key", type="password")
+
+# Pipedream URLs
+st.sidebar.subheader("Pipedream URLs")
+youtube_pipedream_url = st.sidebar.text_input("YouTube URL (Required)", 
+    placeholder="Enter your Pipedream YouTube URL")
+
+# Secondary tool selection
+secondary_tool = st.sidebar.radio(
+    "Select Secondary Tool:",
+    ["Drive", "Notion"]
+)
+
+# Secondary tool URL input
+if secondary_tool == "Drive":
+    drive_pipedream_url = st.sidebar.text_input("Drive URL", 
+        placeholder="Enter your Pipedream Drive URL")
+    notion_pipedream_url = None
+else:
+    notion_pipedream_url = st.sidebar.text_input("Notion URL", 
+        placeholder="Enter your Pipedream Notion URL")
+    drive_pipedream_url = None
+
+# Quick guide before goal input
+st.info("""
+**Quick Guide:**
+1. Enter your Google API key and YouTube URL (required)
+2. Select and configure your secondary tool (Drive or Notion)
+3. Enter a clear learning goal, for example:
+    - "I want to learn python basics in 3 days"
+    - "I want to learn data science basics in 10 days"
+""")
+
+# Main content area
+st.header("Enter Your Goal")
+user_goal = st.text_input("Enter your learning goal:",
+                        help="Describe what you want to learn, and we'll generate a structured path using YouTube content and your selected tool.")
+
+# Progress area
+progress_container = st.container()
+progress_bar = st.empty()
+
+def update_progress(message: str):
+    """Update progress in the Streamlit UI"""
+    st.session_state.current_step = message
+    
+    # Determine section and update progress
+    if "Setting up agent with tools" in message:
+        section = "Setup"
+        st.session_state.progress = 0.1
+    elif "Added Google Drive integration" in message or "Added Notion integration" in message:
+        section = "Integration"
+        st.session_state.progress = 0.2
+    elif "Creating AI agent" in message:
+        section = "Setup"
+        st.session_state.progress = 0.3
+    elif "Generating your learning path" in message:
+        section = "Generation"
+        st.session_state.progress = 0.5
+    elif "Learning path generation complete" in message:
+        section = "Complete"
+        st.session_state.progress = 1.0
+        st.session_state.is_generating = False
+    else:
+        section = st.session_state.last_section or "Progress"
+    
+    st.session_state.last_section = section
+    
+    # Show progress bar
+    progress_bar.progress(st.session_state.progress)
+    
+    # Update progress container with current status
+    with progress_container:
+        # Show section header if it changed
+        if section != st.session_state.last_section and section != "Complete":
+            st.write(f"**{section}**")
+        
+        # Show message with tick for completed steps
+        if message == "Learning path generation complete!":
+            st.success("All steps completed! 🎉")
+        else:
+            prefix = "✓" if st.session_state.progress >= 0.5 else "→"
+            st.write(f"{prefix} {message}")
+
+# Generate Learning Path button
+if st.button("Generate Learning Path", type="primary", disabled=st.session_state.is_generating):
+    if not google_api_key:
+        st.error("Please enter your Google API key in the sidebar.")
+    elif not youtube_pipedream_url:
+        st.error("YouTube URL is required. Please enter your Pipedream YouTube URL in the sidebar.")
+    elif (secondary_tool == "Drive" and not drive_pipedream_url) or (secondary_tool == "Notion" and not notion_pipedream_url):
+        st.error(f"Please enter your Pipedream {secondary_tool} URL in the sidebar.")
+    elif not user_goal:
+        st.warning("Please enter your learning goal.")
+    else:
+        try:
+            # Set generating flag
+            st.session_state.is_generating = True
+            
+            # Reset progress
+            st.session_state.current_step = ""
+            st.session_state.progress = 0
+            st.session_state.last_section = ""
+            
+            result = run_agent_sync(
+                google_api_key=google_api_key,
+                youtube_pipedream_url=youtube_pipedream_url,
+                drive_pipedream_url=drive_pipedream_url,
+                notion_pipedream_url=notion_pipedream_url,
+                user_goal=user_goal,
+                progress_callback=update_progress
+            )
+            
+            # Display results
+            st.header("Your Learning Path")
+            # print(result)
+            if result and "messages" in result:
+                for msg in result["messages"]:
+                    st.markdown(f"📚 {msg.content}")
+
+            else:
+                st.error("No results were generated. Please try again.")
+                st.session_state.is_generating = False
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+            st.error("Please check your API keys and URLs, and try again.")
+            st.session_state.is_generating = False