Update app.py

app.py

@@ -52,44 +52,83 @@ def fetch_message_from_newsweb(url: str) -> str:
 @tool 
 def newsworthiness_detection(fetch_message_from_newsweb: str, newsworthiness_prompt: str) -> str:
     """
-    The returned string from the fetch_message_from_newsweb function is used with the newsworthiness_prompt to define if the message is newsworthy or not
-    
+    Determines the newsworthiness of a stock market announcement using a predefined prompt.
+
     Args:
-        fetch_message_from_newsweb (str): the returned message from the fetch_message_from_newsweb function.
-        newsworthiness_prompt (str): the prompt used to analyze the newsworthiness of the returned message.
+        fetch_message_from_newsweb (str): The stock market message content or a URL to the announcement.
+        newsworthiness_prompt (str): A predefined prompt that provides guidance for assessing the newsworthiness of the announcement.
 
     Returns:
-        str: A grading of newswortiness, either "HØY", "MEDIUM" or "LAV", and a short evaluation of the decision.
+        str: A rating of newsworthiness ("HØY", "MEDIUM", or "LAV") along with a short explanation of the decision.
     """
+    
     newsworthiness_prompt = """
-        Du er en børsjournalist som er ekspert på finansielle markeder og spesielt på Oslo Børs. Du jobber for den norske finansavisen E24. 
-        Jobben din er å gå gjennom alle børsmeldinger fra selskaper notert på Oslo Børs og vurdere hvor egnet meldingen er til å påvirke selskapets aksjekurs. Derfor er du alltid på jakt etter nyhetspoenger i meldingene. Disse kan være av positiv art, som nye kontrakter, pengeinnsprøytning, positive testresultater eller lignende, men ofte er det negative faktorer som påvirker aksjekursen, som nedskrivninger, konflikt, resultatvarsel. Ofte er negative nyheter gjemt nederst i børsmeldingene.
-        Under ser du stikkord, som ofte, men ikke alltid, er egnet til å påvirke markedsverdien til et selskap:
-        
-        Transaction proposal
-        Contemplated 
-        Private placement
-        Rights issue
-        Profit warning
-        Divident 
-        Contract update
-        Transaction
-        Contract
-        Matching halt
-        Inside information
-        Block sale
-        million
-        billion
-        Stepping down
-        Resign
-        Disclosure of large shareholding
-        Strategic review
-        Strategic process
-        
-        Inputen din vil være en børsmelding, enten en url til meldingen eller hele teksten limt inn. Outputen skal være strukturert og inneholde følgende info: Message_id, tittel på børsmeldingen, vurdering av nyhetsverdi, begrunnelse for satt nyhetsverdi.
-        Om det er stor sannsynlighet for at meldingen vil påvirke aksjemarkedet, skal du vurdere at meldingen har HØY nyhetsverdi. Om det er usikkert om meldingen vil påvirke aksjemarkedet, vil du vurdere at meldingen har MEDIUM nyhetsverdi. Om det er lav sannsynlighet for at meldingen vil påvirke aksjemarkedet, vil du vurdere at medlingen har LAV nyhetsverdi.
-        """                      
+    Du er en børsjournalist som er ekspert på finansielle markeder og spesielt på Oslo Børs. Du jobber for den norske finansavisen E24. 
+    Jobben din er å gå gjennom alle børsmeldinger fra selskaper notert på Oslo Børs og vurdere hvor egnet meldingen er til å påvirke selskapets aksjekurs. 
+    Derfor er du alltid på jakt etter nyhetspoenger i meldingene. Disse kan være av positiv art, som nye kontrakter, pengeinnsprøytning, positive testresultater eller lignende, 
+    men ofte er det negative faktorer som påvirker aksjekursen, som nedskrivninger, konflikt, resultatvarsel. Ofte er negative nyheter gjemt nederst i børsmeldingene.
+
+    Under ser du stikkord, som ofte, men ikke alltid, er egnet til å påvirke markedsverdien til et selskap:
+    - Transaction proposal
+    - Contemplated 
+    - Private placement
+    - Rights issue
+    - Profit warning
+    - Dividend
+    - Contract update
+    - Transaction
+    - Contract
+    - Matching halt
+    - Inside information
+    - Block sale
+    - Million
+    - Billion
+    - Stepping down
+    - Resign
+    - Disclosure of large shareholding
+    - Strategic review
+    - Strategic process
+
+    Inputen din vil være en børsmelding, enten en URL til meldingen eller hele teksten limt inn. 
+    Outputen skal være strukturert og inneholde følgende info: **Message_id, tittel på børsmeldingen, vurdering av nyhetsverdi, begrunnelse for satt nyhetsverdi.**
+
+    - Om det er stor sannsynlighet for at meldingen vil påvirke aksjemarkedet, skal du vurdere at meldingen har **HØY nyhetsverdi**.  
+    - Om det er usikkert om meldingen vil påvirke aksjemarkedet, vil du vurdere at meldingen har **MEDIUM nyhetsverdi**.  
+    - Om det er lav sannsynlighet for at meldingen vil påvirke aksjemarkedet, vil du vurdere at meldingen har **LAV nyhetsverdi**.
+    """            
+
+    full_prompt = f"""{newsworthiness_prompt}
+
+    --- Børsmelding ---
+    {fetch_message_from_newsweb}
+    
+    Basert på instruksjonene ovenfor, vurder nyhetsverdien av denne meldingen og gi en begrunnelse. 
+    Svar kun med en av følgende alternativer:
+    - HØY: [begrunnelse]
+    - MEDIUM: [begrunnelse]
+    - LAV: [begrunnelse]
+    """
 
+   # Define API request payload
+    payload = {
+        "inputs": full_prompt,
+        "parameters": {
+            "max_new_tokens": 150,  # Limit response length
+            "temperature": 0.2,  # Keep responses deterministic
+            "return_full_text": False  # Only return generated text
+        }
+    }
+
+    # Send request to Hugging Face API
+    response = requests.post(API_URL, headers=HEADERS, json=payload)
+
+    # Handle errors
+    if response.status_code != 200:
+        return f"Error {response.status_code}: {response.text}"
+
+    # Extract and return response text
+    result = response.json()
+    return result[0]["generated_text"].strip()  # Extracts generated text from response
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: