Update app.py

app.py

@@ -39,28 +39,64 @@ def generate_mba_content(topic, api_key):
         if 'client' in locals():
             del client
 
+# def humanize_text(text, api_key):
+#     if not HUMANIZE_PROMPT:
+#         return "Error: Humanize prompt not found. Ensure 'HUMANIZE_PROMPT' is set in your environment."
+    
+#     try:
+#         client = Groq(api_key=api_key)
+#     except Exception as e:
+#         return f"Error: Failed to initialize Groq client. Check your API key. Details: {str(e)}"
+
+#     prompt = HUMANIZE_PROMPT.replace("[TEXT]", text)
+#     try:
+#         response = client.chat.completions.create(
+#             model="llama3-70b-8192",
+#             messages=[
+#                 {"role": "system", "content": prompt},
+#                 {"role": "user", "content": f"Rewrite the following text: {text}"}
+#             ],
+#             temperature=0.7,
+#             max_tokens=4000,
+#         )
+#         content = response.choices[0].message.content
+#         content = content.replace("—", ",")  # Replace em dashes with commas
+#         if not content.startswith("#"):
+#             content = markdown.markdown(content)
+#         return content
+#     except Exception as e:
+#         return f"Error: Failed to humanize text. Details: {str(e)}"
+#     finally:
+#         api_key = None
+#         if 'client' in locals():
+#             del client
+
 def humanize_text(text, api_key):
+    SYSTEM_PROMPT = "You are a MBA teacher as well as a content writer. Your task is to rewrite provided content to sound more natural and human while following instructions."
+    # HUMANIZE_PROMPT = os.environ.get("HUMANIZE_CONTENT_2")  # Ex: "Rewrite the following text to be more human: [TEXT]"
+
     if not HUMANIZE_PROMPT:
         return "Error: Humanize prompt not found. Ensure 'HUMANIZE_PROMPT' is set in your environment."
-    
+
     try:
         client = Groq(api_key=api_key)
     except Exception as e:
         return f"Error: Failed to initialize Groq client. Check your API key. Details: {str(e)}"
 
-    prompt = HUMANIZE_PROMPT.replace("[TEXT]", text)
+    # Place the input text into the prompt template for the user message
+    user_prompt = HUMANIZE_PROMPT.replace("[TEXT]", text)
     try:
         response = client.chat.completions.create(
             model="llama3-70b-8192",
             messages=[
-                {"role": "system", "content": prompt},
-                {"role": "user", "content": f"Rewrite the following text: {text}"}
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": user_prompt}
             ],
             temperature=0.7,
             max_tokens=4000,
         )
         content = response.choices[0].message.content
-        content = content.replace("—", ",")  # Replace em dashes with commas
+        content = content.replace("—", ",")
         if not content.startswith("#"):
             content = markdown.markdown(content)
         return content