UPDATE: increased timeout

Dockerfile

@@ -28,4 +28,4 @@ RUN mkdir -p /.EasyOCR && chmod -R 777 /.EasyOCR
 
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--timeout-keep-alive", "300", "--timeout-graceful-shutdown", "600"]

app.py

@@ -63,23 +63,11 @@ async def addPDFData(vectorstore: str, pdf: UploadFile = File(...)):
         }
 
 
-@app.post("/addImagePDF")
-async def addPDFData(vectorstore: str, pdf: UploadFile = File(...)):
+@app.post("/scanAndReturnText")
+async def returnText(pdf: UploadFile = File(...)):
     pdf = await pdf.read()
     text = getTextFromImagePDF(pdfBytes = pdf)
-    username, chatbotname = vectorstore.split("-")[1], vectorstore.split("-")[2]
-    df = pd.DataFrame(client.table("ConversAI_ChatbotInfo").select("*").execute().data)
-    currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
-    limit = client.table("ConversAI_UserConfig").select("tokenLimit").eq("username", username).execute().data[0]["tokenLimit"]
-    newCount = currentCount + len(text)
-    if newCount < int(limit):
-        client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
-        return addDocuments(text = text, vectorstore = vectorstore)
-    else:
-        return {
-            "output": "DOCUMENT EXCEEDING LIMITS, PLEASE TRY WITH A SMALLER DOCUMENT."
-        }
-
+    return text
 
 
 @app.post("/addText")
@@ -98,7 +86,6 @@ async def addText(vectorstore: str, text: str):
         }
 
 
-
 class AddQAPair(BaseModel):
     vectorstore: str
     question: str