Update app.py

app.py

@@ -19,6 +19,7 @@ from huggingface_hub import HfApi
 from huggingface_hub import InferenceClient
 from PIL import Image
 import io
+import ast
 
 app = FastAPI()
 
@@ -219,6 +220,25 @@ async def chat(
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error getting chat results: {e}")
 
+# Define a Pydantic model for the request payload
+class ChatRequest(BaseModel):
+    q: str
+    history: str = "[]"
+    model: str = "gpt-4o-mini"
+    proxy: Optional[str] = None
+
+@app.post("/api/chat-post")
+async def chat(request: ChatRequest):
+    """Perform a text search."""
+    try:
+        with WEBS(proxy=request.proxy) as webs:
+            chat_messages = []
+            chat_messages.extend(ast.literal_eval(request.history))
+            results = webs.chat(keywords=request.q, chat_messages = chat_messages, chat model=request.model)
+            return JSONResponse(content=jsonable_encoder(results))
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error getting chat results: {e}")
+
 def extract_text_from_webpage(html_content):
     """Extracts visible text from HTML content using BeautifulSoup."""
     soup = BeautifulSoup(html_content, "html.parser")