Update app.py

app.py

@@ -76,6 +76,19 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
+def send_message_with_retry(chat_session, content, max_retries=3, initial_backoff=30):
+    for attempt in range(max_retries + 1):
+        try:
+            response = chat_session.send_message(content=content)
+            return response
+        except Exception as e:
+            if attempt == max_retries:
+                raise  # Re-raise the exception if max retries reached
+            backoff_time = initial_backoff * (2**attempt)
+            print(f"Error during LLM call: {e}. Retrying in {backoff_time} seconds (Attempt {attempt + 1}/{max_retries + 1})...")
+            time.sleep(backoff_time)
+    return None # Should not reach here as exception is re-raised if all retries fail
+
 
 def extract_notes(filepath):
     files = [
@@ -91,17 +104,22 @@ def extract_notes(filepath):
             },
         ]
     )
-    response = chat_session.send_message(content=prompt)
+
+    response = send_message_with_retry(chat_session, prompt)
     current_response_text = response.text
     while "```json" in response.text:
-        response = chat_session.send_message("CONTINUE")
+        response = send_message_with_retry(chat_session, "CONTINUE")
         current_response_text += "\n" + response.text
 
     pattern = r"```json(.*?)```"
     matches = re.findall(pattern, current_response_text, re.DOTALL)
     results = []
     for match in matches:
-        results.extend(json.loads(match))
+        try:
+            results.extend(json.loads(match))
+        except json.JSONDecodeError as e:
+            print(f"JSONDecodeError: {e} for match: {match}")
+            # Handle JSON decode error, maybe skip or log the issue
 
     return results