Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 
 app = FastAPI()
 
-# Load GPT-SW3-126M
+# Ladda GPT-SW3-126M
 model_name = "AI-Sweden-Models/gpt-sw3-126m"
 try:
     tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -14,27 +14,32 @@ except Exception as e:
     raise e
 
 @app.post("/parse")
-async def parse_arithmetic_prompt(prompt: str):
-    if not prompt or len(prompt) > 100:
-        return {"error": "Ogiltig eller för lång prompt"}
+async def parse_user_request(request: str):
+    if not request or len(request) > 200:
+        return {"error": "Ogiltig eller för lång begäran"}
     try:
-        # Format prompt as plain text per GPT-SW3 documentation
-        prompt = f"Du är en vänlig och hjälpsam assistent. Svara på följande meddelande på svenska:\n\nMeddelande: {message}\nSvar: "
+        # Prompt för att extrahera vara och attribut
+        prompt = f"""Analysera följande begäran på svenska och extrahera nyckelinformation som vara, färg, pris eller andra attribut. Returnera resultatet som en sträng med formatet "vara:namn,attribut1:värde,attribut2:värde" (bara relevanta attribut, utan tomma fält). Om ingen vara identifieras, returnera "error:ingen vara". Exempel:
+- Input: "Jag vill ha en röd bil" -> "vara:bil,färg:röd"
+- Input: "Jag vill ha en ros, helst billigt" -> "vara:ros,pris:billigt"
+
+Begäran: {request}
+Svar: """
         
-        # Tokenize prompt
-        input_ids = tokenizer(full_prompt, return_tensors="pt").input_ids
+        # Tokenisera prompten
+        input_ids = tokenizer(prompt, return_tensors="pt").input_ids
         
-        # Generate response
+        # Generera svar
         with torch.no_grad():
             output = model.generate(
                 input_ids,
-                max_new_tokens=10,
+                max_new_tokens=20,
                 do_sample=False,
                 pad_token_id=tokenizer.pad_token_id,
                 eos_token_id=tokenizer.eos_token_id
             )
         
-        # Decode and extract result
+        # Dekoda och extrahera svaret
         result = tokenizer.decode(output[0], skip_special_tokens=True).strip()
         result = result.split("Svar: ")[-1].strip()
         
@@ -44,4 +49,4 @@ async def parse_arithmetic_prompt(prompt: str):
 
 @app.get("/")
 async def root():
-    return {"message": "Arithmetic Parser API is running!"}
+    return {"Nytt!"}