Update app.py

app.py

@@ -216,6 +216,34 @@ async def website_summarizer(url: str):
         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 summarization: {e}")
+
+@app.get("/api/ask_website")
+async def ask_website(url: str, question: str):
+    """
+    Asks a question about the content of a given website.
+    """
+    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) > 7500:  # Adjust max_chars based on your needs
+            visible_text = visible_text[:7500] + "..."
+
+        # Construct a prompt for the chat model
+        prompt = f"Based on the following text, answer this question in Detail: [QUESTION] {question}\n\n[TEXT] {visible_text}"
+
+        # Use chat model to get the answer
+        with WEBS() as webs:
+            answer_result = webs.chat(keywords=prompt, model="gpt-3.5")
+
+        # Return the answer result
+        return JSONResponse(content=jsonable_encoder({answer_result}))
+
+    except requests.exceptions.RequestException as e:
+        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/maps")
 async def maps(