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_exponential
+from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
 
 def fetch_github_files(github_url, personal_access_token):
     try:
@@ -60,7 +60,12 @@ 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))
+@retry(
+    stop=stop_after_attempt(3),
+    wait=wait_exponential(multiplier=1, min=4, max=10),
+    retry=retry_if_exception_type(Exception),
+    reraise=True
+)
 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')
@@ -88,7 +93,7 @@ def process_with_gemini(file_content, gemini_api_key):
         return response.text
     except Exception as e:
         print(f"Error in Gemini API call: {str(e)}")
-        raise  # This will trigger a retry
+        raise ValueError(f"Gemini API error: {str(e)}")
 
 def process_input(github_url, personal_access_token, gemini_api_key):
     if not github_url.startswith("https://github.com/"):