Update app.py

app.py

@@ -11,15 +11,13 @@ from io import StringIO  # for redirect_stdout
 from functools import wraps  # for caching
 import contextlib  # for redirect_stdout
 import tldextract
-from langchain.prompts import PromptTemplate
-from langchain.chains import LLMChain
-from langchain_openai import ChatOpenAI
 import requests
 import streamlit as st
 import pandas as pd
 import streamlit.components.v1 as components
 import json
 import os
+from openai import OpenAI
 
 
 # ---------------------------------------------------------------------------- #
@@ -87,8 +85,12 @@ else:
                             ADD AS LEARN MORE LINKS FOR THE FIRST TEXT BLOCK LINKS TO structured data https://wordlift.io/blog/en/entity/structured-data/ and schema.org https://wordlift.io/blog/en/entity/schema-org/ TO PROVIDE ADDITIONAL HELP./n/n
                             YOU ARE WRITING THE REPORT IN HTML USING A TEMPLATE.'''}]
 
+        client = OpenAI(api_key=openai_api_key)
+        
+        # Construct messages for the chat API
+        messages = []
+        messages.extend(prefix_messages)
 
-        llm = ChatOpenAI(model_name="gpt-4o-mini", temperature=0, openai_api_key=openai_api_key)
 
         # Create the prompt template and the run statement when there are NOT issues
         if not _issues and len(_items) > 0:
@@ -158,15 +160,22 @@ else:
                                     "advice", "items", "topics", "issues", "technologies"])
             run_statement = {"advice": _advice, "items": _items,
                             "topics": _topics, "issues": _issues, "technologies": _technologies}
+        
+        # Format the prompt
+        user_message = prompt.format(**run_statement)
+        messages.append({"role": "user", "content": user_message})
+
+        # Make the API call
+        try:
+            response = client.chat.completions.create(
+                model="gpt-4o",
+                messages=messages
+            )
+            out = response.choices[0].message.content
+        except Exception as e:
+            out = f"Sorry, there was an error with the OpenAI API: {e}"
 
-        # Create the LLMChain
-        llm_chain = LLMChain(prompt=prompt, llm=llm)
 
-        # If there are no issues, remove the issues from the prompt
-        prompt_text = prompt.format(**run_statement)
-
-        # Run the LLMChain and return the output
-        out = llm_chain.run(**run_statement)
 
         return out