Updated main.py

main.py

@@ -4,7 +4,9 @@ from pydantic import BaseModel
 from my_agent.agent import build_graph
 import pandas as pd
 from typing import Optional , List
-from my_agent.utils.initial_interaction import BusinessInteractionChatbot
+from my_agent.utils.initial_interaction import IntroductionChatbot
+from my_agent.utils.business_interaction import BusinessInteractionChatbot2
+
 from my_agent.utils.utils import encode_image_to_base64 , generate_final_story, generate_image
 
 
@@ -13,7 +15,8 @@ import json
 # Store brainstorming results per thread_id
 
 app = FastAPI()
-interaction_chatbot = BusinessInteractionChatbot()
+introduction_chatbot = IntroductionChatbot()
+interaction_chatbot = BusinessInteractionChatbot2()
 graph = build_graph()
 
 
@@ -22,14 +25,21 @@ stored_data={}
 class UserMessage(BaseModel):
     message: str
 @app.post("/business-interaction")
-def business_chat(msg: UserMessage):
-    response = interaction_chatbot.chat(msg.message)
-    if interaction_chatbot.is_complete(response):
-        details = interaction_chatbot.extract_details()
+def business_introduction_chat(msg: UserMessage):
+    response = introduction_chatbot.chat(msg.message)
+    if introduction_chatbot.is_complete(response):
+        details = introduction_chatbot.extract_details()
         stored_data['business_details'] = details
         return {"response": response, "business_details": details, "complete": True}
     return {"response": response, "complete": False}
 
+@app.post("/business-interaction2")
+def business_interaction_chat(message: str):
+    stored_data ={}
+    stored_data['business_details']={}
+    response = interaction_chatbot.chat(message , stored_data["business_details"])
+    return {'response': response}
+
 
 
 @app.post("/brainstrom")