Update main.py

main.py

@@ -15,17 +15,6 @@ TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
 SysPromptDefault = "You are an expert AI, complete the given task. Do not add any additional comments."    
 SysPromptList = "You are now in the role of an expert AI who can extract structured information from user request. All elements must be in double quotes. You must respond ONLY with a valid python List. Do not add any additional comments."
 
-# Import FastAPI and other necessary libraries
-
-
-# Define the app
-app = FastAPI()
-
-# Create a Pydantic model to handle the input data
-class TopicInput(BaseModel):
-    user_input: str
-    num_topics: int
-
 @retry(tries=3, delay=1)
 def together_response(message, model = "meta-llama/Llama-3-8b-chat-hf", SysPrompt = SysPromptDefault):
   client = Together(api_key=TOGETHER_API_KEY)
@@ -56,10 +45,8 @@ def json_from_text(text):
         fix_json = JSONFixer()
         return loads(fix_json.fix(json_out).line)
     
-SysPromptDefault = "You are an expert AI, complete the given task. Do not add any additional comments."    
-SysPromptList = "You are now in the role of an expert AI who can extract structured information from user request. All elements must be in double quotes. You must respond ONLY with a valid python List. Do not add any additional comments."
-def generate_topics(user_input,num_topics):
-    prompt = f"""create a list of {num_topics} subtopics to follow for conducting {user_input}, RETURN VALID PYTHON LIST"""
+def generate_topics(user_input,num_topics,previous_query):
+    prompt = f"""create a list of {num_topics} subtopics to follow for conducting {user_input} in the context of {previous_query}, RETURN VALID PYTHON LIST"""
     response_topics = together_response(prompt, model = "meta-llama/Llama-3-8b-chat-hf", SysPrompt = SysPromptList)
     subtopics = json_from_text(response_topics)
     return subtopics
@@ -72,6 +59,16 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Define the app
+app = FastAPI()
+
+# Create a Pydantic model to handle the input data
+class TopicInput(BaseModel):
+    user_input: str
+    num_topics: int
+    previous_query: str
+    
+
 @app.get("/",tags=["Home"])
 def api_home():
     return {'detail': 'Welcome to FastAPI Subtopics API! /n visit https://pvanand-generate-subtopics.hf.space/docs to test'}
@@ -79,6 +76,6 @@ def api_home():
 
 @app.post("/generate_topics/")
 async def create_topics(input: TopicInput):
-    topics = generate_topics(input.user_input, input.num_topics)
+    topics = generate_topics(input.user_input, input.num_topics, input.num_topics.previous_query)
     return {"topics": topics}