Included CI CD

.github/workflows/main.yaml

@@ -0,0 +1,29 @@
+name: FastAPI CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build-and-test:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.11'
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r requirements.txt
+        pip install pytest
+
+    - name: Run tests
+      run: pytest

api/main.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI
+from .routers import orchestration, context_analysis, ideation , human_idea_refining , brainstorm , generate_final_story , generate_image
+
+app = FastAPI()
+
+
+@app.get("/")
+async def root():
+    return {'response':'Hello'}
+
+app.include_router(orchestration.router, prefix="/api")
+app.include_router(context_analysis.router, prefix="/api")
+app.include_router(ideation.router, prefix="/api")
+app.include_router(human_idea_refining.router, prefix="/api")
+app.include_router(brainstorm.router, prefix="/api")
+app.include_router(generate_final_story.router, prefix="/api")
+app.include_router(generate_image.router, prefix="/api")
+

api/routers/brainstorm.py

@@ -0,0 +1,40 @@
+from fastapi import APIRouter
+from api.stored_data import stored_data
+from pydantic import BaseModel
+from typing import Optional
+from src.genai.brainstroming_agent.agent import brainstroming_graph
+
+router=APIRouter()
+brainstrom_graph = brainstroming_graph()
+
+
+class BrainstormRequest(BaseModel):
+    preferred_topics: Optional[list] = []
+    image_base64_list: Optional[list] = []
+    thread_id: Optional[str]="default-session"
+
+@router.post("/brainstorm")
+def brainstroming_endpoint(
+    request: BrainstormRequest,  # 🔥 Full JSON body here
+):
+    idea = (
+    [stored_data['refined_ideation']]
+    if stored_data.get('refined_ideation')
+    else [str(stored_data['final_ideation'])]
+    if stored_data.get('final_ideation')
+    else ['''I don't have any idea right now. Create your own **very creative** and **out of the box** video idea and generate the story for now.'''])
+    
+    result = brainstrom_graph.invoke({
+        'idea': idea,
+        'images': request.image_base64_list,
+        'latest_preferred_topics': request.preferred_topics,
+        'business_details': stored_data['business_details']
+    },
+    config={"configurable": {"thread_id": request.thread_id}})
+    
+    stored_data['brainstroming_response'] = result
+
+    return {'response':{
+        "story": result['stories'][-1],
+        "brainstorming_topics": result['brainstroming_topics'][-1]
+    }}

api/routers/context_analysis.py

@@ -0,0 +1,28 @@
+import ast
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from api.stored_data import stored_data
+from src.genai.context_analysis_agent.agent import IntroductionChatbot
+router = APIRouter()
+
+
+class UserMessage(BaseModel):
+    message: str
+context_analysis_graph = IntroductionChatbot()   
+
+@router.post("/context-analysis")
+def context_analysis(msg: UserMessage):
+    response = context_analysis_graph.chat(msg.message)
+    if context_analysis_graph.is_complete(response):
+        details = context_analysis_graph.extract_details()
+        if type(details) != dict:
+            details = details.model_dump()
+        print('Business_details:',details)
+        if isinstance(details, str):
+            details= ast.literal_eval(details)
+        print('Details Type:',type(details))
+        # save_to_db(details)
+        stored_data['business_details'] = details
+        context_analysis_graph.reset()
+        return {"response": response, "business_details": details, "complete": True}
+    return {"response": response, "complete": False}

api/routers/generate_final_story.py

@@ -0,0 +1,15 @@
+from fastapi import APIRouter
+from api.stored_data import stored_data
+from src.genai.brainstroming_agent.utils.utils import generate_final_story
+
+router= APIRouter()
+
+@router.post("/generate-final-story")
+def generate_final_story_endpoint():
+    final_story = generate_final_story(
+    stored_data.get("brainstroming_response") or stored_data.get("business_details")
+    )
+    stored_data['final_story']=final_story
+    return {
+        'response': final_story
+    }

api/routers/generate_image.py

@@ -0,0 +1,14 @@
+from fastapi import APIRouter
+from api.stored_data import stored_data
+from src.genai.brainstroming_agent.utils.utils import generate_image
+router = APIRouter()
+
+@router.post("/generate-image") 
+def generate_image_endpoint():
+    image = generate_image(str(stored_data.get('final_story','''I don't have any story right now. Just use the business details for now.'''))
+                           ,str(stored_data.get('business_details'))
+                           ,str(stored_data.get('refined_ideation','''I don't have any idea right now. Just use the business details for now.''')))
+    stored_data['generated_image']=image
+    return {
+        'response':image
+    }

api/routers/human_idea_refining.py

@@ -0,0 +1,25 @@
+from fastapi import APIRouter
+from pydantic import BaseModel
+from typing import Optional
+from api.stored_data import stored_data
+from src.genai.human_refined_ideation.agent import human_refined_idea
+router= APIRouter()
+human_refine_graph = human_refined_idea()
+
+class RefineIdeationRequest(BaseModel):
+    query: str
+    thread_id: Optional[str]="refine_ideas_thread"
+
+@router.post("/human-idea-refining")
+def human_idea_refine_endpoint(request:RefineIdeationRequest):
+    stored_data['human_ideation_interactions'].append({"role": "user", "content": request.query})
+    response = human_refine_graph.invoke(
+        {
+        'query': stored_data['human_ideation_interactions'],
+        'business_details': stored_data["business_details"],
+        'final_ideation': stored_data.get('final_ideation',["","","",""]),
+        },config={"configurable": {"thread_id": request.thread_id}}
+    )
+    stored_data['human_ideation_interactions'].append({"role": "assistant", "content": response['result']})
+    stored_data['refined_ideation'] = stored_data['human_ideation_interactions'][-1]['content']
+    return {'response' : stored_data['human_ideation_interactions'][-1]['content'] }

api/routers/ideation.py

@@ -0,0 +1,28 @@
+import ast
+from fastapi import APIRouter
+from api.stored_data import stored_data
+from src.genai.ideation_agent.agent import ideation_graph
+from langgraph.errors import GraphRecursionError
+
+router = APIRouter()
+idea_graph = ideation_graph()
+
+@router.post("/ideation")
+def ideation_endpoint():
+    config={"recursion_limit":15, "configurable": {"thread_id": "ideation_thread123"}}
+    try:
+        result = idea_graph.invoke(
+        {
+            'business_details': [stored_data['business_details']],
+            'image_caption': [stored_data['image_caption'] if 'image_caption' in stored_data else ""]
+
+        },
+        config=config,
+        )
+        stored_data['final_ideation'] = result['improver_response'][-1]
+        stored_data['final_ideation']=ast.literal_eval(stored_data['final_ideation'])
+
+        return {'response':ast.literal_eval(result['improver_response'][-1])}
+    except GraphRecursionError:
+        result = idea_graph.get_state({"configurable": {"thread_id": "ideation_thread123"}})
+        return {'response': result[0]}

api/routers/orchestration.py

@@ -0,0 +1,30 @@
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from api.stored_data import stored_data
+from src.genai.orchestration_agent.agent import orchestration_chat
+from typing import Optional
+
+class OrchestrationRequest(BaseModel):
+    message: str
+    image_base64 : Optional[list] = []
+
+router= APIRouter()
+
+@router.post("/orchestration")
+def orchestration_endpoint(request:OrchestrationRequest):
+    print('Image:',request.image_base64)
+    result = orchestration_chat(request.message , request.image_base64)
+    if result.image_caption != '':
+        stored_data['image_caption']=result.image_caption
+    if result.video_idea !='' or result.video_idea != 'null':
+        stored_data['refined_ideation']= result.video_idea
+    if result.video_story!='' or result.video_story!='null':
+        stored_data['final_story']= result.video_story
+    print('Idea:',stored_data['refined_ideation'])
+    print('Story:', stored_data['final_story'])
+
+    return {'tool_response': result.tool ,
+             'message_response': result.query_response, 
+             'image_caption':result.image_caption,
+             'video_idea': result.video_idea,
+             'video_story': result.video_story}

api/routers/show_analytics.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter
+from api.stored_data import stored_data
+from src.genai.orchestration_agent.utils.utils import show_analytics
+
+router=APIRouter()
+@router.post("/show-analytics")
+def show_analytics_endpoint():
+    response = show_analytics(stored_data['business_details'])
+    return {'response': response}

api/stored_data.py

@@ -0,0 +1,10 @@
+stored_data = {}
+stored_data['business_details'] = {
+    "business_type": "fitness and gym",
+    "platform": "Instagram, TikTok",
+    "target_audience": "young Nepali adults (ages 18–40) who are health-conscious and active on social media",
+    "business_goals": "to expand gym branches across all major cities of Nepal and build a strong fitness community",
+    "offerings": "personal training, group fitness classes, modern workout equipment, nutrition guidance, and wellness programs",
+    "Challenges_faced": "attracting loyal members, standing out in a competitive market, and promoting consistent engagement"
+}
+stored_data['human_ideation_interactions'] = []

main.py

@@ -1,20 +1,20 @@
 from fastapi import FastAPI , UploadFile , File , Form 
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
-from brainstroming_agent.agent import brainstroming_graph
+from src.genai.brainstroming_agent.agent import brainstroming_graph
 import pandas as pd
 from typing import Optional , List
-from context_analysis_agent.agent import IntroductionChatbot
-from business_interaction_agent.agent import BusinessInteractionChatbot
-from context_analysis_agent.utils.utils import save_to_db
+from src.genai.context_analysis_agent.agent import IntroductionChatbot
+from src.genai.business_interaction_agent.agent import BusinessInteractionChatbot
+from src.genai.context_analysis_agent.utils.utils import save_to_db
 import ast
-from orchestration_agent.agent import orchestration_chat
-from orchestration_agent.utils.utils import caption_image , show_analytics
-from brainstroming_agent.utils.utils import encode_image_to_base64 , generate_final_story, generate_image
-from idea_to_budget_agent.agent import budget_calculator
-from ideation_agent.agent import ideation_graph
+from src.genai.orchestration_agent.agent import orchestration_chat
+from src.genai.orchestration_agent.utils.utils import caption_image , show_analytics
+from src.genai.brainstroming_agent.utils.utils import encode_image_to_base64 , generate_final_story, generate_image
+from src.genai.idea_to_budget_agent.agent import budget_calculator
+from src.genai.ideation_agent.agent import ideation_graph
 from langgraph.errors import GraphRecursionError
-from human_refined_ideation.agent import human_refined_idea
+from src.genai.human_refined_ideation.agent import human_refined_idea
 from dummy_state import stored_data, long_term_memory
 import json
 

requirements.txt

@@ -21,4 +21,4 @@ pandas
 langchain-community
 tiktoken
 langchain-anthropic
-
+pytest