Update main.py

main.py

@@ -79,8 +79,7 @@ class URLQuery(BaseModel):
 @app.post("/chat-with-url")
 async def chat_with_url(data: URLQuery):
     try:
-        os.environ["USER_AGENT"] = "Mozilla/5.0"
-        loader = WebBaseLoader(data.url)
+        loader = WebBaseLoader(data.url, header_template={"User-Agent": "Mozilla/5.0"})
         documents = loader.load()
         web_content = "\n".join([doc.page_content for doc in documents])
 
@@ -94,16 +93,18 @@ async def chat_with_url(data: URLQuery):
         return {"answer": response.choices[0].message.content}
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
+        
 
 @app.post("/extract-text-from-image")
 async def extract_text_from_image(file: UploadFile = File(...)):
     try:
         contents = await file.read()
-        image = Image.open(io.BytesIO(contents))
+        image = Image.open(io.BytesIO(contents)).convert("RGB")
         text = pytesseract.image_to_string(image)
         return {"answer": text.strip() or "⚠️ No text extracted."}
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
+        
 
 @app.post("/transcribe-audio")
 async def transcribe_audio(file: UploadFile = File(...)):