Update app.py

app.py

@@ -244,7 +244,92 @@ async def ask_website(url: str, question: str, model: str = "llama-3-70b"):
         raise HTTPException(status_code=500, detail=f"Error fetching or processing URL: {e}")
     except Exception as e:
         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."""
+    try:
+        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
+
+@app.get("/api/web_gemini")
+async def web_gemini(
+    q: str,
+    model: str = "flash"
+):
+    """Perform a text search."""
+    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:
+                    link = result['href']
+                    try:
+                        response = requests.get(link, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/111.0"})
+                        response.raise_for_status()
+                        visible_text = extract_text_from_webpage(response.text)
+                        if len(visible_text) > 10000:
+                            visible_text = visible_text[:10000] + "..."
+                        extracted_results.append({"link": link, "text": visible_text})
+                    except requests.exceptions.RequestException as e:
+                        print(f"Error fetching or processing {link}: {e}")
+                        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:
+            data = response.json()
+            return data['content']
+        else:
+            print(f"API request failed with status code: {response.status_code}")
+            return None
+
+@app.get("/api/ask_website_gemini")
+async def ask_website_gemini(
+    url: str,
+    q: str,
+    model: str = "flash"
+):
+    """Perform a text search."""
+    try:
+        # Extract text from the given URL
+        response = requests.get(url, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/111.0"})
+        response.raise_for_status()
+        visible_text = extract_text_from_webpage(response.text)
+        if len(visible_text) > 30000:  # Adjust max_chars based on your needs
+            visible_text = visible_text[:30000] + "..."        
+        if model=="pro":        
+            url = f"https://gemini-pro.developer-house.workers.dev/?question= Based on the following text, answer this question in Paragraph: [QUESTION] {question} [TEXT] {visible_text}"
+        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
+
+            
 @app.get("/api/maps")
 async def maps(
     q: str,