Update app.py

app.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import JSONResponse, StreamingResponse
-from webscout import WEBS, YTTranscriber, LLM
+from webscout import WEBS, YTTranscriber, LLM, GoogleS
 from typing import Optional, List, Dict
 from fastapi.encoders import jsonable_encoder
 from bs4 import BeautifulSoup
@@ -13,9 +13,9 @@ from huggingface_hub import InferenceClient
 from PIL import Image
 import io
 from easygoogletranslate import EasyGoogleTranslate
-
 from pydantic import BaseModel
 
+
 app = FastAPI()
 
 # Define Pydantic models for request payloads
@@ -427,7 +427,59 @@ async def adv_web_search(
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during advanced search: {e}")
+@app.post("/api/AI_search_google")
+async def adv_web_search(
+    q: str,
+    model: str = "gpt-4o-mini",  # Use webs.chat by default
+    max_results: int = 5,
+    timelimit: Optional[str] = None,
+    safesearch: str = "moderate",
+    region: str = "wt-wt",
+    # backend: str = "html",
+    max_chars: int = 15000,
+    system_prompt: str = "You are an advanced AI chatbot. Provide the best answer to the user based on Google search results.",
+    proxy: Optional[str] = None
+):
+    """
+    Combines web search, web extraction, and chat model for advanced search.
+    """
+    try:
+        with GoogleS(proxy=proxy) as webs:
+            search_results = webs.search(query=q, region=region,
+                                     safe=safesearch,
+                                     time_period=timelimit,
+                                     max_results=max_results)
+            # 2. Extract text from top search result URLs asynchronously
+            extracted_text = ""
+            tasks = [fetch_and_extract(result['href'], 6000, proxy) for result in search_results if 'href' in result]
+            extracted_results = await asyncio.gather(*tasks)
+            for result in extracted_results:
+                if result['text'] and len(extracted_text) < max_chars:
+                    extracted_text += f"## Content from: {result['link']}\n\n{result['text']}\n\n"
+
+            extracted_text[:max_chars]
+
+
+        # 3. Construct the prompt for the chat model
+        ai_prompt = (
+            f"User Query: {q}\n\n"
+            f"Please provide a detailed and accurate answer to the user's query. Include relevant information extracted from the search results below. Ensure to cite sources by providing links to the original content where applicable. Format your response as follows:\n\n"
+            f"1. **Answer:** Provide a clear and comprehensive answer to the user's query.\n"
+            f"2. **Details:** Include any additional relevant details or explanations.\n"
+            f"3. **Summary:** Provide a summary of the search results. **"
+            f"4. **Sources:** List the sources of the information with clickable links for further reading.\n\n"
+            f"Search Results:\n{extracted_text}"
+        )
+
+        # 4. Get the chat model's response using webs.chat 
+        with WEBS(proxy=proxy) as webs:
+            response = webs.chat(keywords=ai_prompt, model=model)
+
+        # 5. Return the results
+        return JSONResponse(content={"answer": response})
 
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error during advanced search: {e}")
 
 @app.get("/api/website_summarizer")
 async def website_summarizer(url: str, proxy: Optional[str] = None):
@@ -483,22 +535,7 @@ async def ask_website(url: str, question: str, model: str = "llama-3-70b", proxy
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during question answering: {e}")
 
-# Stable Diffusion client
-client_sd3 = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
 
-@app.get("/api/sd3")
-def sd3(prompt: str = "", steps: int = 20, width: int = 1000, height: int = 1000):
-    try:
-        image = client_sd3.text_to_image(
-            prompt=f"{prompt}, hd, high quality, 4k, masterpiece",
-            num_inference_steps=steps,
-            width=width,
-            height=height
-        )
-        image = Image.open(io.BytesIO(image))
-        return image 
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error during image generation: {e}")
 
 @app.get("/api/translate")
 async def translate(
@@ -562,4 +599,4 @@ def get_ascii_weather(location: str):
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8083)
+    uvicorn.run(app, host="0.0.0.0", port=8083)