Update app.py

app.py

@@ -3,6 +3,7 @@ import google.generativeai as genai
 import requests
 import base64
 import json
+from tenacity import retry, stop_after_attempt, wait_exponential
 
 def fetch_github_files(github_url, personal_access_token):
     try:
@@ -59,10 +60,16 @@ def fetch_github_files(github_url, personal_access_token):
     except json.JSONDecodeError:
         return f"Error: Unable to parse GitHub API response for {file_path}"
 
+@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=4, max=10))
 def process_with_gemini(file_content, gemini_api_key):
     genai.configure(api_key=gemini_api_key)
     model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
     
+    # Limit content size (adjust the limit as needed)
+    max_content_length = 10000  # characters
+    if len(file_content) > max_content_length:
+        file_content = file_content[:max_content_length] + "..."
+    
     prompt = f"""
     Analyze the following file content for open-source license information:
 
@@ -76,8 +83,12 @@ def process_with_gemini(file_content, 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, timeout=60)  # Set a timeout of 60 seconds
+        return response.text
+    except Exception as e:
+        print(f"Error in Gemini API call: {str(e)}")
+        raise  # This will trigger a retry
 
 def process_input(github_url, personal_access_token, gemini_api_key):
     if not github_url.startswith("https://github.com/"):