Update app.py

app.py

@@ -1,16 +1,18 @@
 import streamlit as st
 from pubmed_rag import search_pubmed, fetch_pubmed_abstracts, summarize_text
 from image_pipeline import analyze_medical_image
-from models import chat_with_openai
+from models import query_medical_text
 
 st.set_page_config(page_title="Advanced Medical AI", layout="wide")
 
+
 def main():
     st.title("Advanced Medical AI")
     st.sidebar.title("Features")
     task = st.sidebar.selectbox("Choose a task:", ["PubMed Q&A", "Medical Image Analysis"])
 
     if task == "PubMed Q&A":
+        # PubMed Q&A Section
         st.subheader("PubMed Question Answering")
         query = st.text_input("Enter your medical question:", "What are the latest treatments for diabetes?")
         max_results = st.slider("Number of PubMed articles to retrieve:", 1, 10, 5)
@@ -30,22 +32,28 @@ def main():
             for pmid, summary in summaries.items():
                 st.write(f"**PMID {pmid}**: {summary}")
 
-            system_message = "You are a medical assistant with access to PubMed summaries."
-            user_message = query
-            with st.spinner("Generating answer..."):
-                answer = chat_with_openai(system_message, user_message)
+            with st.spinner("Querying the medical reasoning model..."):
+                answer = query_medical_text(query)
             st.subheader("AI-Powered Answer")
             st.write(answer)
 
     elif task == "Medical Image Analysis":
+        # Medical Image Analysis Section
         st.subheader("Medical Image Analysis")
         uploaded_file = st.file_uploader("Upload a medical image (PNG/JPG):", type=["png", "jpg", "jpeg"])
+
         if uploaded_file:
+            # Display the uploaded image
             st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+            
+            # Analyze the medical image
             with st.spinner("Analyzing image..."):
                 result = analyze_medical_image(uploaded_file)
+            
+            # Display the result
             st.subheader("Diagnostic Insight")
             st.write(result)
 
+
 if __name__ == "__main__":
-    main()
+    main()