Update main.py

main.py

@@ -135,32 +135,35 @@ def json_from_text(text):
     """
     Extracts JSON from text using regex and fuzzy JSON loading.
     """
-    match = re.search(r'\{[\s\S]*\}', text)
-    if match:
-      json_out = match.group(0)
-    else:
-      json_out = text
     try:
-        # Using fuzzy json loader
-        return loads(json_out)
-    except Exception:
-        # Using JSON fixer/ Fixes even half json/ Remove if you need an exception
-        fix_json = JSONFixer()
-        return loads(fix_json.fix(json_out).line)
-    
+      return json.loads(text)
+    except:
+      match = re.search(r'\{[\s\S]*\}', text)
+      if match:
+        json_out = match.group(0)
+      else:
+        json_out = text
+      # Use Fuzzy JSON loading
+      return loads(json_out)
+
+@retry(tries=3, delay=0.5)
 def generate_topics(user_input, num_topics, previous_queries):
     prompt = prompt_topics.format(user_input=user_input, num_topics=num_topics)
     response_topics = together_response(prompt, model=llm_default_medium, SysPrompt=SysPromptList, temperature=1)
     subtopics = json_from_text(response_topics)
     return subtopics
 
+@retry(tries=3, delay=0.5)
 def generate_subtopics(main_task,user_input,num_topics,excluded_topics):
+    print("trying")
+    1/0
     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=llm_default_medium, SysPrompt=SysPromptList, temperature=1)
     subtopics = json_from_text(response_topics)
     return subtopics
 
+@retry(tries=3, delay=0.5)
 def generate_report(topic, description):
     prompt = f"""create a detailed report on: {topic} by following the instructions: {description}"""
     md_report = together_response(prompt, model = llm_default_medium, SysPrompt = SysPromptMdOffline)