Update app.py

app.py

@@ -1,9 +1,49 @@
 import streamlit as st
 
+
+def get_diabetic_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 
+    personalize heading hook line for article and need only article heading nothing else
+
+    article content: {text} """
+    PROMPT = PromptTemplate(
+        template=prompt_template, input_variables=["text"]
+    )
+    chain = LLMChain(llm=llm, prompt=PROMPT)
+    resp = chain.run(text=data)
+
+    return resp
+
+
+def get_diabetic_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 
+    personalize heading hook line for article and need only article heading nothing else
+
+    article content: {text} """
+    PROMPT = PromptTemplate(
+        template=prompt_template, input_variables=["text"]
+    )
+    chain = LLMChain(llm=llm, prompt=PROMPT)
+    resp = chain.run(text=data)
+
+    return resp
+
+
 def process_article_content(content):
-    words = len(content.split())
-    characters = len(content)
-    return words, characters
+    heading_1 = get_diabetic_heading(content)
+    heading_2 = get_cancer_heading(content)
+    
+
+    
+    # words = len(content.split())
+    # characters = len(content)
+    return heading_1, heading_2
 
 # Streamlit app
 def main():