Update app.py

app.py

@@ -65,8 +65,10 @@ def analyze_article(text: str) -> dict:
     "{safe_text}"
     """
     
-    response = client.text_generation(prompt, max_new_tokens=250, temperature=0.1)
-    llm_data = _extract_json_from_llm(response)
+    messages = [{"role": "user", "content": prompt}]
+    response = client.chat_completion(messages=messages, max_tokens=250, temperature=0.1)
+    response_text = response.choices[0].message.content
+    llm_data = _extract_json_from_llm(response_text)
 
     subjectivity_score = TextBlob(safe_text).sentiment.subjectivity
     raw_reading_ease = textstat.flesch_reading_ease(safe_text)
@@ -213,8 +215,10 @@ def check_contradiction(text_a: str, text_b: str) -> dict:
     Text 1: "{text_a[:1000]}"
     Text 2: "{text_b[:1000]}"
     """
-    response = client.text_generation(prompt, max_new_tokens=100, temperature=0.1)
-    result = _extract_json_from_llm(response)
+    messages = [{"role": "user", "content": prompt}]
+    response = client.chat_completion(messages=messages, max_tokens=100, temperature=0.1)
+    response_text = response.choices[0].message.content
+    result = _extract_json_from_llm(response_text)
     return {"relationship": result.get("relationship", "NEUTRAL"), "confidence": result.get("confidence", 0.0)}