Update app.py

app.py

@@ -218,7 +218,7 @@ async def website_summarizer(url: str):
         raise HTTPException(status_code=500, detail=f"Error during summarization: {e}")
 
 @app.get("/api/ask_website")
-async def ask_website(url: str, question: str):
+async def ask_website(url: str, question: str, model: str = "llama-3-70b"):
     """
     Asks a question about the content of a given website.
     """
@@ -231,11 +231,11 @@ async def ask_website(url: str, question: str):
             visible_text = visible_text[:7500] + "..."
 
         # Construct a prompt for the chat model
-        prompt = f"Based on the following text, answer this question in Paragraph: [QUESTION] {question}\n\n[TEXT] {visible_text}"
+        prompt = f"Based on the following text, answer this question in Paragraph: [QUESTION] {question} [TEXT] {visible_text}"
 
         # Use chat model to get the answer
         with WEBS() as webs:
-            answer_result = webs.chat(keywords=prompt, model="gpt-3.5")
+            answer_result = webs.chat(keywords=prompt, model=model)
 
         # Return the answer result
         return JSONResponse(content=jsonable_encoder({answer_result}))