Update: Tools create and analyze, TechTool, session, init

src/streamlit_app.py

@@ -1,40 +1,89 @@
-import altair as alt
-import numpy as np
-import pandas as pd
+# import phase
 import streamlit as st
+import pydantic import BaseModel, Field
+from typing import List, Literal, Dict, Optional
+from agency_swarn import Agent, Agency, set_openai_key, BaseTool
 
-"""
-# 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])),
-    ))
+class AnalyzeProjectRequirement(BaseTool):
+    project_name: str = Field(..., description = "Name of Project")
+    project_description: str = Field(..., description = "Project Description and Goals")
+    project_type: Literal["Web Application", "Mobile App", "API Development", "Data Analytics", "AI/ML Solution", "Other"] = Field(..., description = "Type of project")
+    budget_range: Literal["$10k-$25k", "$25k-$50k", "$100k+"] = Field(..., description = "Budget Range of the project")
+
+    class ToolConfig:
+        name = "analyze_project"
+        description = "Analyze project requirements and feasibility"
+        one_call_at_a_time = True
+
+    def run(self) -> str:
+        """Analyze project and store results in shared state"""
+        if self._shared_state.get("project_analysis", None) is not None:
+            raise ValueError('Project Analysis already exists, you can move forward to technical specification.')
+
+        analysis = {
+            "name": self.project_name,
+            "type": self.project_type,
+            "complexity": "high",
+            "timeline": "6 months",
+            "budget_feasibility": "within range",
+            "requirements": ["Scalable Architecture", "Security", "API Integration"]
+        }
+
+        self._shared_state.set("project_analysis", analysis)
+        return "Project analysis completed. Please proceed with technical specification"
+
+class CreateTechnicalSpecification(BaseTool):
+    core_technologies: str = Field(..., description = "Comma-seperated list of main technologies and frameworks")
+    architecture_type: Literal["monolithic", "microservice", "serverless", "hybrid"] = Field(..., description = "Proposed Architecture Type")
+    scalability_requirement: Literal("high", "medium", "low") = Field(..., description = "Scalability Needs")
+
+    class ToolConfig:
+        name = "create_technical_spec"
+        description = "Creates technical specifications based on project analysis"
+        one_call_at_a_time = True
+
+    def run(self) -> str:
+        """Create technical specification based on the analysis"""
+        project_analysis = self._shared_state.get("project_analysis", None)
+        if project_analysis is None:
+            raise ValueError("Please analyse project requirements first using the AnalyzeProjectRequirement tool")
+
+        spec ={
+            "project_name" : project_analysis["name"],
+            "architecture": self.architecture_type,
+            "technologies": self.core_technologies.split(","), 
+            "scalability": self.scalability_requirement
+        }
+        self._shared_state.set("technical_specification", spec)
+        return f"Technical Specification created for {project_analysis['name']}"
+
+def init_session_state() -> None:
+    """Initiate session state variable"""
+    if "messages" is not st.session_state:
+        st.session_state.message = []
+    if "api_key" not in st.session_state:
+        st.session_state.api_key = None
+
+
+def main() -> None:
+    st.set_page_config(page_title = "dev.ai - Place for innovation", layout = "wide")
+    init_session_state()
+
+    st.title("🎯 dev.ai services - Innovate, Build and Lighten up the future with AI")
+
+    # API Configuration
+    with st.sidebar:
+        st.header("API Configuration")
+        openai_api_key = st.text_input(
+            "OpenAI API Key",
+            type = "password",
+            help = "Enter your OpenAI Key that start with sk--xxxxxxxxxxxxxx"
+        )
+
+        if openai_api_key:
+            st.session_state.api_key = openai_api_key
+            st.success("API Key accepted")
+        else:
+            st.warning("⚠️ API Key not found, please enter you key to proceed")
+            st.markdown("[Get your API key here]("https://platform.openai.com/api-keys)")
+            return