Update app.py

app.py

@@ -1,47 +1,57 @@
 import streamlit as st
-
-# Assuming ai_doctor_chat is a custom function you've defined in ai_assistant.py
 from ai_assistant import ai_doctor_chat
 
-# Display title with centralized HTML styling
+# Display title
 st.markdown("<h1 style='text-align: center;'>Your AI Doctor Using Your Custom Knowledge Base &#129302;</h1>", unsafe_allow_html=True)
 
-# Create layout with two columns, adjusting the ratio for better visual distribution
+# Create layout with two columns
 left_column, right_column = st.columns([1, 3])
 
-# Display an image in the left column. Assuming 'ai_doctor_img.jpg' is correctly located in your app's directory.
-left_column.image("ai_doctor_img.jpg", width=200, use_column_width=True)
+# Display image in the left column
+left_column.image("ai_doctor_img.jpg", width=200, use_column_width="auto")
 
-# Create a text input box for the OpenAI key, ensuring security by masking input
+# Create a text input box for the OpenAI key
 openai_key = right_column.text_input('Enter your OpenAI Key', type='password')
 
-# Create a text input for user queries
-query = right_column.text_input('Enter your query')
-# Button to submit the query
+query = right_column.text_input('Enter your query', type='default')
 submit = right_column.button('Submit')
-
 if submit:
     if query and openai_key:
         try:
-            # Display a spinner while processing the query to improve user experience
             with st.spinner('Processing your query...'):
                 response = ai_doctor_chat(openai_key, query)
-            right_column.success(response)
+            right_column.write(response)
         except Exception as e:
-            right_column.error(f'An error occurred: {e}')
+            right_column.error(f'An error occurred: {e}', icon='🚨')
     else:
-        right_column.error('Please enter both your OpenAI key and your query!', icon="🚨")
+        right_column.error('Please enter your OpenAI key and Query both!', icon="🚨")
 
-# Separator for visual clarity
 st.markdown("---")
 st.write("Connect with me:")
-
-# Create a row of columns for social media links
+# Define the URLs for the images you want to display
+image_urls = {
+        "kaggle": "https://www.kaggle.com/static/images/site-logo.svg",
+        "linkedin": "URL_TO_YOUR_LINKEDIN_IMAGE",
+        "google_scholar": "URL_TO_YOUR_GOOGLE_SCHOLAR_IMAGE",
+        "youtube": "URL_TO_YOUR_YOUTUBE_IMAGE",
+        "github": "URL_TO_YOUR_GITHUB_IMAGE"
+    }
+
+    # Create columns for each social profile link
 kaggle, linkedin, google_scholar, youtube, github = st.columns(5)
 
-# Each column contains a hyperlink to a social profile, wrapped in markdown for formatting
+    # For each column, display an image and a markdown link to a social profile
+kaggle.image(image_urls["kaggle"], width=100)
 kaggle.markdown("[Kaggle](https://www.kaggle.com/muhammadimran112233)")
+    
+linkedin.image(image_urls["linkedin"], width=100)
 linkedin.markdown("[LinkedIn](https://www.linkedin.com/in/muhammad-imran-zaman)")
+    
+google_scholar.image(image_urls["google_scholar"], width=100)
 google_scholar.markdown("[Google Scholar](https://scholar.google.com/citations?user=ulVFpy8AAAAJ&hl=en)")
+    
+youtube.image(image_urls["youtube"], width=100)
 youtube.markdown("[YouTube](https://www.youtube.com/@consolioo)")
+    
+github.image(image_urls["github"], width=100)
 github.markdown("[GitHub](https://github.com/Imran-ml)")