Update main.py

main.py

@@ -42,6 +42,9 @@ class InputData(BaseModel):
     NumeroCaratteriSplitInstruction: int = 30000
     EliminaRisposteNonPertinenti: bool = False
     UnificaRispostaPertinente: bool = False
+    telegramChatId: str = ''
+    telegramUrlBot: str = ''
+    telegramUrlPost: str = ''
 
 class InputDataAsync(InputData):
     test: str = ''
@@ -72,7 +75,8 @@ def generate_text(request: Request, input_data: InputData):
         max_new_tokens = min(max_new_tokens, 29500 - len(input_text))
         history = []
         generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
-        #generated_response = generated_response.replace('"', '')
+        if input_data.telegramChatId != '' && input_data.telegramUrlBot != '' && input_data.telegramUrlPost != '': 
+            call_telegram_api(input_data, generated_response)
         LoggaTesto("RISPOSTA", {"response": generated_response})
         return {"response": generated_response}
     else: 
@@ -95,7 +99,17 @@ def generate_text(request: Request, input_data: InputData):
             #result_data = result_data.replace('"', '')
             LoggaTesto("RISPOSTA ASINCRONA UNIFICATA", {"response": result_data})
         return {"response": result_data}
-    
+
+def call_telegram_api(input_data, generated_response): 
+    payload = {
+        "chat_id": input_data.telegramChatId,
+        "text": generated_response",
+        "telegramUrl": input_data.telegramUrlBot
+    }
+    response = requests.post(input_data.telegramUrlPost, json=payload)
+    if response.status_code != 200:
+        print("Errore nella richiesta POST. Codice di stato:", response.status_code)    
+
 def generate_input_text(input_data):
     if input_data.instruction.startswith("http"):
         try: