Update app.py

app.py

@@ -1,12 +1,14 @@
 import streamlit as st
-from interm import generate_clinical_content, generate_summary, generate_soap_note, save_as_text, save_as_pdf
+from interm import generate_clinical_content, generate_summary, generate_soap_note, fetch_pubmed_articles, save_as_text, save_as_pdf
+
+# Set page configuration
+st.set_page_config(page_title="Clinical AI Assistant", page_icon="🩺", layout="wide")
 
-# Title & Introduction
 st.title("🩺 AI-Powered Clinical Intelligence Assistant")
-st.write("Revolutionizing **clinical documentation, research summarization, and medical education** using AI.")
+st.write("AI-driven **clinical research, medical documentation, and PubMed insights**.")
 
 # Tabs for different functionalities
-tabs = st.tabs(["📝 Clinical Content", "📑 Research Summaries", "🩺 SOAP Notes", "📊 Medical Reports"])
+tabs = st.tabs(["📝 Clinical Content", "📑 Research Summaries", "🩺 SOAP Notes", "🔍 PubMed Research", "📊 Medical Reports"])
 
 # Tab 1: Generate Clinical Content
 with tabs[0]:
@@ -56,8 +58,27 @@ with tabs[2]:
         else:
             st.warning("Please enter patient symptoms.")
 
-# Tab 4: Download Clinical Reports
+# Tab 4: PubMed Research Integration
 with tabs[3]:
+    st.header("Fetch Latest Research from PubMed")
+
+    query = st.text_input("Enter a medical keyword (e.g., COVID-19, AI in Oncology, Diabetes):")
+
+    if st.button("Fetch PubMed Articles"):
+        if query:
+            with st.spinner("Retrieving PubMed articles..."):
+                articles = fetch_pubmed_articles(query)
+                st.session_state["pubmed_results"] = articles
+                for article in articles:
+                    st.subheader(article["title"])
+                    st.write(f"**Authors:** {article['authors']}")
+                    st.write(f"**Abstract:** {article['abstract']}")
+                    st.write(f"[Read More]({article['url']})")
+        else:
+            st.warning("Please enter a medical topic.")
+
+# Tab 5: Download Clinical Reports
+with tabs[4]:
     st.header("Download Clinical Reports")
 
     if "soap_note" in st.session_state: