Update app.py

app.py

@@ -8,11 +8,11 @@ from langchain.chains import LLMChain
 
 api_key = os.environ['openai_api_key']
 
-def get_diabetic_heading(data):
+def get_insurance_heading(data):
 
     llm = OpenAI(temperature=0,openai_api_key=api_key)
 
-    prompt_template = """this is article content and my user is diabetic health now give me user specific or we can 
+    prompt_template = """this is article content and my user is insurance now give me user specific or we can 
     personalize heading hook line for article and need only article heading nothing else
 
     article content: {text} """
@@ -25,11 +25,11 @@ def get_diabetic_heading(data):
     return resp
 
 
-def get_cancer_heading(data):
+def get_sports_heading(data):
 
     llm = OpenAI(temperature=0,openai_api_key=api_key)
 
-    prompt_template = """this is article content and my user is cancer health now give me user specific or we can 
+    prompt_template = """this is article content and my user is sports now give me user specific or we can 
     personalize heading hook line for article and need only article heading nothing else
 
     article content: {text} """
@@ -44,7 +44,7 @@ def get_cancer_heading(data):
 
 def process_article_content(content):
 
-    return get_diabetic_heading(content), get_cancer_heading(content)
+    return get_insurance_heading(content), get_sports_heading(content)
 
 # Streamlit app
 def main():
@@ -57,12 +57,12 @@ def main():
     if st.button("Process"):
         # Process the article content
         if article_content:
-            diabetic_user, cancer_user = process_article_content(article_content)
+            insurance_user, sports_user = process_article_content(article_content)
 
             # Display the output
             st.subheader("Processed Output:")
-            st.write(f"{diabetic_user}")
-            st.write(f"{cancer_user}")
+            st.write(f"{insurance_user}")
+            st.write(f"{sports_user}")
 
 if __name__ == "__main__":
     main()