Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from styles import apply_styling
-from utils import remove_reasoning_and_sources, clean_explanation, format_follow_up_questions
+from utils import remove_reasoning_and_sources, clean_explanation
 from session_state import initialize_session_state, add_message_to_history, get_full_history
 from chat_display import display_chat_history, show_typing_indicator, display_legal_disclaimer
 from model import (
@@ -48,100 +48,3 @@ with st.sidebar:
             # Reset report step for a new report
             st.session_state.report_step = 1
             st.rerun()
-    
-    # # Add End Conversation button
-    # if st.button("End Conversation", use_container_width=True):
-    #     end_conversation()
-    #     st.rerun()
-    
-    # Show report form when button is clicked - only if not processing conversation
-    if st.session_state.get('show_report_form', False) and not st.session_state.get('processing', False) and not st.session_state.get('conversation_lock', False):
-        st.subheader("Report Information")
-        generate_and_download_report()
-
-# Main app area
-st.title("How are you Feeling Today?")
-
-# Display chat history
-display_chat_history()
-
-# Display typing indicator if processing
-show_typing_indicator()
-
-# Chat input
-if prompt := st.chat_input("Describe your symptoms or ask a medical question..."):
-    # Set conversation lock to prioritize conversation over report generation
-    st.session_state.conversation_lock = True
-    
-    # Add user message to history using the database-backed function
-    add_message_to_history({"role": "user", "content": prompt})
-    
-    # Set processing flag to true
-    st.session_state.processing = True
-    
-    # Force refresh to show the typing indicator
-    st.rerun()
-
-# Check if we need to process a response (this block runs after the rerun if processing is True)
-if st.session_state.processing:
-    try:
-        # Get the full history from database
-        full_history = get_full_history()
-        
-        # Get the last user message
-        last_user_message = next((msg["content"] for msg in reversed(full_history) 
-                                if msg["role"] == "user"), None)
-        
-        if last_user_message:
-            # Process as normal conversation
-            # Filter out the last user message as it will be processed separately
-            previous_messages = [msg for msg in full_history if msg["role"] == "assistant" or 
-                               (msg["role"] == "user" and msg["content"] != last_user_message)]
-            
-            # Updated to handle the new response format with follow-up questions and sources
-            reply, explanation, follow_up_questions, sources, evidence = orchestrator_chat(
-                previous_messages,
-                last_user_message,
-                use_rag=st.session_state.use_rag,
-                is_follow_up=len(previous_messages) > 0
-            )
-            
-            # Clean up the response and explanation
-            cleaned_reply = remove_reasoning_and_sources(reply) if reply else ""
-            cleaned_explanation = clean_explanation(explanation) if explanation else ""
-            
-            # Format follow-up questions for better display
-            formatted_follow_up = format_follow_up_questions(follow_up_questions) if follow_up_questions else ""
-            
-            # Add assistant response to history with the new fields
-            assistant_message = {
-                "role": "assistant", 
-                "content": cleaned_reply,
-                "explanation": cleaned_explanation,
-                "follow_up_questions": formatted_follow_up,
-                "sources": sources,
-                "evidence": evidence if evidence else []
-            }
-            
-            # Save to database-backed history
-            add_message_to_history(assistant_message)
-    finally:
-        # Set processing back to false regardless of success/failure
-        st.session_state.processing = False
-        # Release conversation lock after response is processed
-        st.session_state.conversation_lock = False
-        
-    # Force refresh to update the UI with the response
-    st.rerun()
-
-# Small, unobtrusive legal disclaimer 
-display_legal_disclaimer()
-
-# This ensures app.py can be used as a direct entry point for deployment platforms like Hugging Face
-if __name__ == "__main__":
-    # Add the current directory to the Python path if needed
-    import os
-    import sys
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    if current_dir not in sys.path:
-        sys.path.insert(0, current_dir)