design changes

app.py

@@ -303,6 +303,10 @@ with st.expander("**What is FOOD CHAIN?**"):
         Made by the Classify Crew: [Contact List](https://linktr.ee/classifycrew)
         """
     )
+    if st.theme() == 'Dark':
+        st.image('logo-dark.png', caption="Uploaded Image.", use_container_width=True)
+    else:
+        st.image('logo-light.png', caption="Uploaded Image.", use_container_width=True)
 #################
 
 # Image Classification Section
@@ -319,7 +323,7 @@ if uploaded_image and query:
         fpredictions = ""
     
         # Show the top predictions with percentages
-        st.write("Top Predictions:")
+        st.markdown("**Top Predictions:**")
         for class_name, confidence in predictions:
             if int(confidence) > 0.05:
                 fpredictions += f"{class_name}: {confidence:.2f}%,"
@@ -351,11 +355,11 @@ if uploaded_image and query:
     openAIresponse = llm.invoke(openAICall)
     print("AI CALL RESPONSE: ", openAIresponse.content)
 
+    RAGresponse = get_response(openAIresponse.content + " " + query)
     with st.expander("Recipe Generation", expanded=True, icon=':material/menu_book:'):
         st.title('Results: RAG')
         # RAG the openai response and display
         print("RAG INPUT", openAIresponse.content + " " + query)
-        RAGresponse = get_response(openAIresponse.content + " " + query)
         display_response(RAGresponse)
 elif uploaded_image is not None:
     with st.expander("**Food Classification**", expanded=True, icon=':material/search_insights:'):
@@ -371,7 +375,7 @@ elif uploaded_image is not None:
         fpredictions = ""
     
         # Show the top predictions with percentages
-        st.write("Top Predictions:")
+        st.markdown("**Top Predictions:**")
         for class_name, confidence in predictions:
             if int(confidence) > 0.05:
                 fpredictions += f"{class_name}: {confidence:.2f}%,"
@@ -382,9 +386,9 @@ elif uploaded_image is not None:
         print(fpredictions)
 
 elif query:
-     with st.expander("**Recipe Generation**", expanded=True, icon=':material/menu_book:'):
+    response = get_response(query)
+    with st.expander("**Recipe Generation**", expanded=True, icon=':material/menu_book:'):
         st.title("Results: RAG")
-        response = get_response(query)
         display_response(response)
 else:
     st.warning("Please input an image and/or a prompt.", icon=':material/no_meals:')