Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,70 @@
-import altair as alt
-import numpy as np
-import pandas as pd
-import streamlit as st
-
-"""
-# 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])),
-    ))
+import streamlit as st 
+from pydantic import BaseModel, Field
+from typing import List, Literal, Dict, Optional 
+from agency_swarn import Agent, Agency, set_openai_key, basetool
+
+
+class AnalyzeProjectRequirement(BaseTool):
+    project_name: str= Field(...., description = " Name of Project")
+    project_description: str= Field(...., description= "Project and Goals")
+    project_type: Literal ["Web Application", "Mobile App", "API Developement", "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 requirement 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 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 to technical specifications"
+
+
+
+
+    class CreateTechnicalSPecification(BaseTool):
+        core_technologies: str = Field(.., description= "Comma-separated list of main technologies")
+        architecture_type: Literal["monolithic", "microservice", "serverless", "hybrid"] = Field(...., description= "Proposed Architecture")
+        scalability_requirement: Literal("high","medium", "low") = Field(...., description= "Scalability Needs")
+        
+    
+
+    class ToolConfig:
+        name= "create_techinal_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"""
+        if self._shared_state.get("project_analysis")
+        if self._shared_state.get("project Analysis already exists, you can move forward to technical specification.")
+
+        spec= {
+            "project_name": project_analysis["name"],
+            "architecture": self.architecture_type,
+            "technologies": self.core_technologies.split(","),
+            "scalability": self.scalability_requirement
+        }
+
+        self._shared_state.set("project_analysis", analysis)
+        return "Project Analysis completed. Please proceed to technical specifications"
+
+
+
+def init_session_state()-> None:
+    """Initiate session state variable"""
+    if "messages" is not st.session_state
+    
+