Update app.py

app.py

@@ -30,11 +30,18 @@ agent = initialize_agent(
 def azure_cert_bot(cert_name):
     query = f"Microsoft Azure {cert_name} certification curriculum site:microsoft.com"
     search_results = ddgs.run(query).split("\n")
-    prompt = f"Based on the following curriculum details, generate key questions and answers in markdown format for the {cert_name} certification exam:\n{search_results}"
+    prompt = f"Based on the following curriculum details, generate key questions and answers in markdown format for the {cert_name} certification exam. Do not include any metadata or unnecessary text, only return the formatted Q&A:\n{search_results}"
     response = llm.invoke(prompt)
     
-    return search_results, response
+    try:
+        response_text = response.get("content", "No response generated.") if isinstance(response, dict) else response
+        response_text = "\n".join([line for line in response_text.split("\n") if not line.lower().startswith("content=") and "metadata" not in line.lower()])
+    except Exception as e:
+        response_text = f"Error processing response: {str(e)}"
+    
+    return search_results, response_text
 
+# Streamlit UI Enhancements
 st.set_page_config(page_title="Azure Certification Prep Assistant", layout="wide")
 
 # Custom Styling
@@ -82,4 +89,3 @@ if st.button("Get Certification Details"):
         st.markdown(f"<div class='markdown-text-container'>{qa_content}</div>", unsafe_allow_html=True)
     else:
         st.warning("Please enter a certification name.")
-