Update app.py

app.py

@@ -245,20 +245,44 @@ async def ask_website(url: str, question: str, model: str = "llama-3-70b"):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during question answering: {e}")
 
+import requests
+
+def get_gemini_content(question, model="flash"):
+    """
+    Sends a request to the Gemini API and returns the content of the response.
+
+    Args:
+        question: The question to ask the Gemini API.
+        model: The Gemini model to use (flash or pro). Defaults to "flash".
+
+    Returns:
+        The content string from the API response, or None if the request fails.
+    """
+
+    if model == "pro":
+        url = f"https://gemini-pro.developer-house.workers.dev/?question={question}"
+    else:
+        url = f"https://gemini-flash.developer-house.workers.dev/?question={question}"
+    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
+
 @app.get("/api/gemini")
 async def gemini(q: str, model: str = "flash"):
     """Get answers from Gemini models."""
     try:
-        if model == "pro":
-            url = f"https://gemini-pro.developer-house.workers.dev/?question={q}"
+        content = get_gemini_content(q, model)
+        if content:
+            return JSONResponse(content=jsonable_encoder({"answer": content})) 
         else:
-            url = f"https://gemini-flash.developer-house.workers.dev/?question={q}"
-        response = requests.get(url)
-        response.raise_for_status()  # Raise an error for bad status codes
-        data = response.json()
-        return JSONResponse(content=jsonable_encoder({data['content']})) # Use JSONResponse to return a JSON
-    except requests.exceptions.RequestException as e:
-        raise HTTPException(status_code=500, detail=f"Gemini API request failed: {e}")
+            raise HTTPException(status_code=500, detail="Gemini API request failed.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error during Gemini request: {e}")