Update main.py

main.py

@@ -130,6 +130,13 @@ def generate_topics(user_input, num_topics, previous_queries):
     subtopics = json_from_text(response_topics)
     return subtopics
 
+def generate_subtopics(main_task,user_input,num_topics,excluded_topics):
+    excluded_topics = ",".join(excluded_topics)
+    prompt = prompt_subtopics.format(main_task = main_task,user_input=user_input, num_topics=num_topics, excluded_topics=excluded_topics)
+    response_topics = together_response(prompt, model="meta-llama/Llama-3-70b-chat-hf", SysPrompt=SysPromptList, temperature=1)
+    subtopics = json_from_text(response_topics)
+    return subtopics
+
 def generate_report(topic, description):
     prompt = f"""create a detailed report on: {topic} by following the instructions: {description}"""
     md_report = together_response(prompt, model = "meta-llama/Llama-3-70b-chat-hf", SysPrompt = SysPromptMdOffline)