Update app.py

app.py

@@ -246,34 +246,26 @@ async def ask_website(url: str, question: str, model: str = "llama-3-70b"):
         raise HTTPException(status_code=500, detail=f"Error during question answering: {e}")
 
 @app.get("/api/gemini")
-async def gemini(
-    q: str,
-    model: str = "flash"
-):
-    """Perform a text search."""
+async def gemini(q: str, model: str = "flash"):
+    """Get answers from Gemini models."""
     try:
-        if model=="pro":        
+        if model == "pro":
             url = f"https://gemini-pro.developer-house.workers.dev/?question={q}"
         else:
             url = f"https://gemini-flash.developer-house.workers.dev/?question={q}"
         response = requests.get(url)
-        if response.status_code == 200:
-            data = response.json()
-            return data['content']
-        else:
-            print(f"API request failed with status code: {response.status_code}")
-            return None
+        response.raise_for_status()  # Raise an error for bad status codes
+        data = response.json()
+        return data['content']
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Gemini API request failed: {e}")
 
 @app.get("/api/web_gemini")
-async def web_gemini(
-    q: str,
-    model: str = "flash"
-):
-    """Perform a text search."""
+async def web_gemini(q: str, model: str = "flash"):
+    """Search the web and get answers from Gemini models."""
     try:
         with WEBS() as webs:
             search_results = webs.text(keywords=q, max_results=3)
-            # Extract text from each result's link
             extracted_results = []
             for result in search_results:
                 if 'href' in result:
@@ -290,18 +282,22 @@ async def web_gemini(
                         extracted_results.append({"link": link, "text": None})
                 else:
                     extracted_results.append({"link": None, "text": None})
-                content=jsonable_encoder({extracted_results})        
-        if model=="pro":        
-            url = f"https://gemini-pro.developer-house.workers.dev/?question= Based on the following text from Google Search, answer this question in Paragraph: [QUESTION] {question} [TEXT] {content}"
-        else:
-            url = f"https://gemini-flash.developer-house.workers.dev/?question= Based on the following text from Google Search, answer this question in Paragraph: [QUESTION] {question} [TEXT] {content}"
-        response = requests.get(url)
-        if response.status_code == 200:
+
+            content = jsonable_encoder({"extracted_results": extracted_results})
+
+            # Call Gemini API based on the selected model
+            if model == "pro":
+                url = f"https://gemini-pro.developer-house.workers.dev/?question=Based on the following text from Google Search, answer this question in Paragraph: [QUESTION] {q} [TEXT] {content}"
+            else:
+                url = f"https://gemini-flash.developer-house.workers.dev/?question=Based on the following text from Google Search, answer this question in Paragraph: [QUESTION] {q} [TEXT] {content}"
+
+            response = requests.get(url)
+            response.raise_for_status()
             data = response.json()
             return data['content']
-        else:
-            print(f"API request failed with status code: {response.status_code}")
-            return None
+
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Error during web search or Gemini request: {e}")
 
 @app.get("/api/ask_website_gemini")
 async def ask_website_gemini(
@@ -322,14 +318,11 @@ async def ask_website_gemini(
         else:
             url = f"https://gemini-flash.developer-house.workers.dev/?question={q}"
         response = requests.get(url)
-        if response.status_code == 200:
-            data = response.json()
-            return data['content']
-        else:
-            print(f"API request failed with status code: {response.status_code}")
-            return None
+        data = response.json()
+        return data['content']
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Gemini API request failed: {e}")
 
-            
 @app.get("/api/maps")
 async def maps(
     q: str,