Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import re
 from groq import Groq  
 
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
@@ -20,14 +21,14 @@ def call_groq_api(specific, measurable, achievable, relevant, time_bound):
     
     chat_completion = groq_client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
-        model="llama2-70b-4096",
+        model="llama3-70b-8192",
     )
     
     if chat_completion.choices:
         generated_text = chat_completion.choices[0].message.content
         print("LLM Response:", generated_text)  
-        options = generated_text.split('\n\n') 
-        formatted_options = [opt for opt in options if opt.strip().startswith('Option')]
+        options = re.split(r'\n\s*\n', generated_text)
+        formatted_options = [opt for opt in options if opt.strip().startswith('**Option')]
         return '\n\n'.join(formatted_options)
     else:
         print("Error: No response from Groq API or API returned an unexpected format.")
@@ -85,7 +86,7 @@ interface = gr.Interface(
     outputs=[
         gr.Textbox(label="Generated SMART Goals", lines=10, placeholder="Generated content will appear here...")
     ],
-    title="SMART Goals Builder powered by Llama2 and Groq",
+    title="SMART Goals Builder powered by Llama3 and Groq",
     description="Fill out each field to generate SMART goals."
 )