Update app.py

app.py

@@ -3,7 +3,7 @@ import google.generativeai as genai
 import requests
 import base64
 import json
-from tenacity import retry, stop_after_attempt, wait_fixed
+from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
 
 def fetch_github_files(github_url, personal_access_token):
     try:
@@ -77,19 +77,27 @@ def process_chunk_with_gemini(chunk, gemini_api_key):
     5. Provide no other information such as greeting or summary as the purpose is to catalog and document all open source licenses used.
     """
 
-    response = model.generate_content(prompt)
-    return response.text
+    try:
+        response = model.generate_content(prompt)
+        return response.text
+    except Exception as e:
+        print(f"Error processing chunk: {str(e)}")
+        return f"Error processing chunk: {str(e)}"
 
-@retry(stop=stop_after_attempt(3), wait=wait_fixed(2))
+@retry(stop=stop_after_attempt(5), wait=wait_exponential(multiplier=1, min=4, max=10), retry=retry_if_exception_type(Exception))
 def process_with_gemini(file_content, gemini_api_key):
-    # Split the content into chunks of approximately 4000 characters
-    chunk_size = 4000
+    # Split the content into smaller chunks
+    chunk_size = 2000
     chunks = [file_content[i:i+chunk_size] for i in range(0, len(file_content), chunk_size)]
     
     results = []
     for chunk in chunks:
-        result = process_chunk_with_gemini(chunk, gemini_api_key)
-        results.append(result)
+        try:
+            result = process_chunk_with_gemini(chunk, gemini_api_key)
+            results.append(result)
+        except Exception as e:
+            print(f"Error processing chunk: {str(e)}")
+            results.append(f"Error processing chunk: {str(e)}")
     
     # Combine the results
     combined_result = "\n\n".join(results)