Update main.py

main.py

@@ -52,26 +52,36 @@ def read_root(request: Request, input_data: InputData):
     max_new_tokens = input_data.max_new_tokens
     top_p = input_data.top_p
     repetition_penalty = input_data.repetition_penalty
+    if input_data.instruction.startswith("http"):
+        try:
+            resp = requests.get(input_data.instruction)
+            resp.raise_for_status()  # Lancia un'eccezione per errori HTTP
+            input_data.instruction = resp.text
+        except requests.exceptions.RequestException as e:
+            input_data.instruction = ""
     history = [] 
-    input_text = f'''
-    {{
-        "input": {{
-            "role": "system",
-            "content": "{input_data.systemRole}", 
-            "style": "{input_data.systemStyle}"
-        }},
-        "messages": [
-            {{
-                "role": "instructions",
-                "content": "{input_data.instruction}"
+    if input_data.systemRole != "" or input_data.systemStyle != "" or input_data.instruction != ""
+        input_text = f'''
+        {{
+            "input": {{
+                "role": "system",
+                "content": "{input_data.systemRole}", 
+                "style": "{input_data.systemStyle}"
             }},
-            {{
-                "role": "user",
-                "content": "{input_data.input}"
-            }}
-        ]
-    }}
-    '''
+            "messages": [
+                {{
+                    "role": "instructions",
+                    "content": "{input_data.instruction}"
+                }},
+                {{
+                    "role": "user",
+                    "content": "{input_data.input}"
+                }}
+            ]
+        }}
+        '''
+    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}")