Update main.py

main.py

@@ -46,9 +46,10 @@ def json_from_text(text):
         fix_json = JSONFixer()
         return loads(fix_json.fix(json_out).line)
     
-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)
+def generate_topics(user_input, num_topics, previous_queries):
+    previous_context = " -> ".join(previous_queries)
+    prompt = f"""create a list of {num_topics} subtopics to follow for conducting {user_input} in the context of {previous_context}, RETURN VALID PYTHON LIST"""
+    response_topics = together_response(prompt, model="meta-llama/Llama-3-8b-chat-hf", SysPrompt=None)
     subtopics = json_from_text(response_topics)
     return subtopics
 
@@ -67,15 +68,14 @@ app.add_middleware(
 class TopicInput(BaseModel):
     user_input: str = Query(default="market research", description="input query to generate subtopics")
     num_topics: int = Query(default=5, description="Number of subtopics to generate (default: 5)")
-    previous_query: str = Query(default="", description="Previous query for context (default: empty string)")
+    previous_queries: list[str] = Query(default=[], description="List of previous queries for context")
 
-@app.get("/",tags=["Home"])
+@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'}
-
+    return {'detail': 'Welcome to FastAPI Subtopics API! Visit https://pvanand-generate-subtopics.hf.space/docs to test'}
 
 @app.post("/generate_topics")
 async def create_topics(input: TopicInput):
-    topics = generate_topics(input.user_input, input.num_topics, input.previous_query)
+    topics = generate_topics(input.user_input, input.num_topics, input.previous_queries)
     return {"topics": topics}