Upload app.py

app.py

@@ -11,6 +11,7 @@ from langchain_community.chat_models import ChatOllama
 from langchain_groq import ChatGroq
 import logging
 from analyzers import DrugInteractionAnalyzer
+import base64
 
 # Load environment variables
 load_dotenv()
@@ -127,33 +128,111 @@ def create_agent_graph(owl_file_path: str) -> StateGraph:
     return workflow.compile()
 
 def main():
-    st.title("Drug Interaction Analysis System")
+    st.set_page_config(page_title="OntoGraph", page_icon="💊", layout="wide")
 
-    user_input = st.text_input("Enter drug names separated by commas (e.g., Aspirin, Warfarin):", value="")
-
-    if st.button("Analyze"):
-        if not user_input.strip():
-            st.warning("Please enter at least one drug name.")
-            return
+    st.markdown("<h1 style='text-align: center;'>OntoGraph - Drug Interaction Analysis System 🧬</h1>", unsafe_allow_html=True)
+    
+    st.markdown("<p style='text-align: center; color: #c9f9fa;'>This application uses a combination of ontology-based reasoning and language models to analyze drug interactions.</p>", unsafe_allow_html=True)
+    st.markdown("<p style='text-align: center; color: #faffb7;'>Analyze potential interactions between different drugs</p>", unsafe_allow_html=True)
+        
+    st.markdown("""
+                <style>
+                .centered {
+                    display: flex;
+                    flex-direction: column;
+                    align-items: center;
+                    justify-content: center;
+                    text-align: center;
+                }
+                </style>
+                <div class="centered">
+            """, unsafe_allow_html=True)
+
+    def get_base64_image(img_path):
+        with open(img_path, "rb") as f:
+            encoded = base64.b64encode(f.read()).decode()
+        return f"data:image/webp;base64,{encoded}"
+
+    # Generate the image URL
+    image_url = get_base64_image("img/header-image.webp")
+
+    # Load custom CSS with the embedded image
+    st.markdown(f"""
+        <style>
+            .parallax {{
+                background-image: url("{image_url}");
+                min-height: 200px;
+                background-attachment: fixed;
+                background-position: center;
+                background-repeat: no-repeat;
+                background-size: cover;
+            }}
+        </style>
+    """, unsafe_allow_html=True)
+
+    st.markdown('<div class="parallax"></div>', unsafe_allow_html=True) 
+    
+    st.markdown("<br>", unsafe_allow_html=True)
 
-        owl_file_path = os.path.join("ontology", "DrugInteraction.owl")
-        if not os.path.exists(owl_file_path):
-            logging.error(f"Ontology file not found: {owl_file_path}")
-            st.error(f"Ontology file not found: {owl_file_path}")
-            return
+    col1, col2, col3 = st.columns([3, 1, 7])
 
-        try:
-            with st.spinner("Analyzing drug interactions..."):
-                agent_graph = create_agent_graph(owl_file_path)
-                result = agent_graph.invoke(GraphState(input=user_input))
+    with col3:
 
-            st.subheader("Analysis Results:")
-            st.markdown(result["response"])
+        st.markdown("<h2 style='text-align: left; color: #d46c6c;'>Instructions</h2>", unsafe_allow_html=True)
+        st.markdown("<p style='text-align: left; color: white;'>1. Enter the drug names separated by commas. <br> 2. Click on the 'Analyze' button. <br> 3. Wait for the analysis to complete. <br> 4. View the results below.</p>", unsafe_allow_html=True)
+        st.markdown("<div class='divider'></div>", unsafe_allow_html=True)
+        st.markdown("<hr>", unsafe_allow_html=True)
 
-            logging.info("Analysis completed and results displayed.")
-        except Exception as e:
-            logging.error(f"An error occurred: {str(e)}")
-            st.error(f"An error occurred: {str(e)}")
+        st.markdown('<h3 class="big-font">Enter drug names separated by commas (e.g., Aspirin, Warfarin):</h3>', unsafe_allow_html=True)
+        user_input = st.text_input("", value="", key="drug_input")
+        st.markdown('<style>div[data-testid="stTextInput"] input { font-size: 16px;}</style>', unsafe_allow_html=True)
+        
+        if st.button("Analyze"):
+            if not user_input.strip():
+                st.warning("Please enter at least one drug name.")
+                return
+
+            owl_file_path = os.path.join("ontology", "DrugInteraction.owl")
+            if not os.path.exists(owl_file_path):
+                logging.error(f"Ontology file not found: {owl_file_path}")
+                st.error(f"Ontology file not found: {owl_file_path}")
+                return
+
+            try:
+                with st.spinner("Analyzing drug interactions..."):
+                    agent_graph = create_agent_graph(owl_file_path)
+                    result = agent_graph.invoke(GraphState(input=user_input))
+
+                st.subheader("Analysis Results:")
+                st.markdown(result["response"])
+
+                logging.info("Analysis completed and results displayed.")
+            except Exception as e:
+                logging.error(f"An error occurred: {str(e)}")
+                st.error(f"An error occurred: {str(e)}")
+
+    with col1:
+        st.markdown("<h4 style='text-align: center; color: #5ac5c9;'>Comprehensive Drug Analysis</h4>", unsafe_allow_html=True)
+        # st.markdown("<img src='https://cdn.dribbble.com/users/228367/screenshots/4603754/chemistryset_dribbble.gif' alt='reaction' width='100%'/>", unsafe_allow_html=True)
+        st.markdown("<div class='divider'></div>", unsafe_allow_html=True)
+        image_base64 = get_base64_image("img/reaction.gif")
+        st.markdown(f"<img src='{image_base64}' width='100%'/>", unsafe_allow_html=True)
+        st.markdown("<br>", unsafe_allow_html=True)
+        st.write("""
+        - Drug interaction detection
+        - Conflict identification
+        - Similar drug suggestions
+        - Alternative medication recommendations
+                """)
+        
 
+    st.markdown("<hr>", unsafe_allow_html=True)
+    st.markdown("<p style='text-align: center; color: #faffb7;'></p>", unsafe_allow_html=True)
+    st.write("""
+    <div style='text-align: center;'>
+        <p style='text-align: left; color: #d46c6c;'>Disclaimer: </p><p style='text-align: left; color: #707377;'>This application is intended for informational purposes only and does not replace professional medical advice, diagnosis, or treatment. The analysis provided is based on the data available in the ontology and may not account for all possible drug interactions. Users are strongly advised to consult a licensed healthcare provider before making any decisions based on the analysis results. The creators of this application are not responsible for any decisions made or actions taken based on the information provided.</p>
+    </div>
+    """, unsafe_allow_html=True)
+    
 if __name__ == "__main__":
     main()