Update main.py

main.py

@@ -47,11 +47,18 @@ class PostSpazio(BaseModel):
     
 #--------------------------------------------------- Generazione TESTO ------------------------------------------------------
 @app.post("/Genera")
-def read_root(request: Request, input_data: InputData):
+def generate_text(request: Request, input_data: InputData):
     temperature = input_data.temperature
     max_new_tokens = input_data.max_new_tokens
     top_p = input_data.top_p
     repetition_penalty = input_data.repetition_penalty
+    input_text = generate_input_text(input_data)
+    print(f"{datetime.now()} - Input Text: {input_text}")
+    generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
+    print(f"{datetime.now()} - Response Text: {generated_response}")
+    return {"response": generated_response}
+
+def generate_input_text(input_data):
     if input_data.instruction.startswith("http"):
         try:
             resp = requests.get(input_data.instruction)
@@ -60,7 +67,7 @@ def read_root(request: Request, input_data: InputData):
         except requests.exceptions.RequestException as e:
             input_data.instruction = ""
     history = [] 
-    if input_data.systemRole != "" or input_data.systemStyle != "" or input_data.instruction != ""
+    if input_data.systemRole != "" or input_data.systemStyle != "" or input_data.instruction != "":
         input_text = f'''
         {{
             "input": {{
@@ -82,10 +89,7 @@ def read_root(request: Request, input_data: InputData):
         '''
     else:
         input_text = input_data.input   
-    print(f"{datetime.now()} - Input Text: {input_text}")
-    generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
-    print(f"{datetime.now()} - Response Text: {generated_response}")
-    return {"response": generated_response}
+    return input_text
 
 def generate(prompt, history, temperature=0.7, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
     temperature = float(temperature)
@@ -115,7 +119,7 @@ def format_prompt(message, history):
 
 #--------------------------------------------------- Generazione TESTO ASYNC ------------------------------------------------------
 @app.post("/GeneraAsync")
-def read_rootAsync(request: Request, input_data: InputDataAsync):
+def generate_textAsync(request: Request, input_data: InputDataAsync):
     print(input_data.input)
     result_data = asyncio.run(GeneraTestoAsync("https://matteoscript-fastapi.hf.space/Genera", input_data))
     return {"response": result_data}