Update main.py

main.py

@@ -28,6 +28,9 @@ app.add_middleware(
 
 class InputData(BaseModel):
     input: str
+    systemRole: str = '' 
+    systemStyle: str = ''
+    instruction: str = ''
     temperature: float = 0.7
     max_new_tokens: int = 30000
     top_p: float = 0.95
@@ -46,17 +49,35 @@ class PostSpazio(BaseModel):
     nomeSpazio: str
     input: str = '' 
     api_name: str = "/chat"
-
     
 #--------------------------------------------------- Generazione TESTO ------------------------------------------------------
 @app.post("/Genera")
 def read_root(request: Request, input_data: InputData):
-    input_text = input_data.input
     temperature = input_data.temperature
     max_new_tokens = input_data.max_new_tokens
     top_p = input_data.top_p
     repetition_penalty = input_data.repetition_penalty
     history = [] 
+    input_text = = f'''
+        {{
+            "input": {{
+                "role": "system",
+                "content": "{input_data.systemRole}", 
+                "style": "{input_data.systemStyle}", 
+                
+            }},
+            "messages": [
+                {{
+                    "role": "instructions",
+                    "content": {input_data.instruction}
+                }},
+                {{
+                    "role": "user",
+                    "content": "{input_data.input}"
+                }}
+            ]
+        }}
+        '''    
     generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
     return {"response": generated_response}