Update main.py

main.py

@@ -40,6 +40,7 @@ class InputData(BaseModel):
     NumeroGenerazioni: int = 1
     StringaSplit: str = '********'
     NumeroCaratteriSplitInstruction: int = 30000
+    EliminaRisposteNonPertinenti: bool = False
 
 class InputDataAsync(InputData):
     test: str = ''
@@ -75,6 +76,8 @@ def generate_text(request: Request, input_data: InputData):
         input_data.asincrono = False
         result_data = asyncio.run(GeneraTestoAsync("https://matteoscript-fastapi.hf.space/Genera", input_data))
         LoggaTesto("RISPOSTA ASINCRONA FINALE", {"response": result_data})
+        if input_data.EliminaRisposteNonPertinenti and "response" in result_data and "NOTFOUND" in result_data["response"]:
+            del result_data["response"]  
         return {"response": result_data}
     
 def generate_input_text(input_data):
@@ -164,19 +167,6 @@ async def make_request(session, token, data, url, max_retries=3):
             await asyncio.sleep(1)  
     raise Exception("Max retries reached or skipping retries. Unable to make the request.")
 
-
-async def make_request_old(session, token, data, url):
-    headers = {
-        'Content-Type': 'application/json',
-        'Authorization': 'Bearer ' + token
-    }
-    async with session.post(url, headers=headers, json=data) as response:
-        try:
-            result_data = await response.json()
-        except aiohttp.ContentTypeError:
-            result_data = await response.text()   
-        return result_data
-
 async def CreaListaInput(input_data):
     if input_data.instruction.startswith("http"):
         try:
@@ -270,7 +260,6 @@ def DividiInstructionText(input_data):
             ListaInput.append(data)
     return ListaInput
 
-
 async def GeneraTestoAsync(url, input_data):
     token = os.getenv('TOKEN')
     async with aiohttp.ClientSession() as session:
@@ -281,7 +270,6 @@ async def GeneraTestoAsync(url, input_data):
             tasks.extend([make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)])
         return await asyncio.gather(*tasks)
 
-
         
 #--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------
 style_image = {