qna interface bug fix

app.py

@@ -293,6 +293,9 @@ def main():
     /* Feature cards for homepage */
     .feature-card {background-color: rgba(255, 255, 255, 0.1); color: inherit; border-radius: 10px; padding: 20px; min-height: 220px; box-shadow: 0 4px 6px rgba(0,0,0,0.1); display: flex; flex-direction: column; align-items: center;}
     .feature-card p {font-size: 0.9em; margin-top: 5px; max-width: 100%; overflow-wrap: break-word; word-wrap: break-word;}
+    
+    /* Fixed input container */
+    .input-container {margin-bottom: 20px; padding: 15px; border-radius: 10px; background-color: rgba(255, 255, 255, 0.05); border: 1px solid rgba(255, 255, 255, 0.1);}
     </style>
     """, unsafe_allow_html=True)
 
@@ -378,6 +381,10 @@ def main():
 
     # Chat interface
     elif st.session_state.page == 'chat':
+        # Initialize session state for input if not exists
+        if 'user_input' not in st.session_state:
+            st.session_state.user_input = ""
+        
         # Header with clear button
         col1, col2 = st.columns([3, 1])
         with col1:
@@ -386,41 +393,22 @@ def main():
             # Add a clear button in the header
             if st.button("🗑️ Clear Chat"):
                 st.session_state.chat_history = []
+                st.session_state.user_input = ""
                 st.rerun()
-                
+        
         st.markdown("Ask any clinical diagnostic question and get insights based on medical knowledge and patient cases.")
         
-        # Display chat history
-        for i, (query, response) in enumerate(st.session_state.chat_history):
-            st.markdown(f"<div class='chat-message chat-message-user'><b>🧑‍⚕️ You:</b> {query}</div>", unsafe_allow_html=True)
-            
-            st.markdown(f"<div class='chat-message chat-message-assistant'><b>🩺 DiReCT:</b> {response['answer']}</div>", unsafe_allow_html=True)
-            
-            with st.expander("View Sources"):
-                for doc in response["context"]:
-                    st.markdown(f"<div class='source-box'>"
-                              f"<b>Source:</b> {Path(doc.metadata['source']).stem}<br>"
-                              f"<b>Type:</b> {doc.metadata['type']}<br>"
-                              f"<b>Content:</b> {doc.page_content[:300]}...</div>", 
-                              unsafe_allow_html=True)
-            
-            # Show evaluation metrics if available
-            try:
-                eval_scores = evaluate_rag_response(response, embeddings)
-                with st.expander("View Evaluation Metrics"):
-                    col1, col2 = st.columns(2)
-                    with col1:
-                        st.metric("Hit Rate (Top-3)", f"{eval_scores['hit_rate']:.2f}")
-                    with col2:
-                        st.metric("Faithfulness", f"{eval_scores['faithfulness']:.2f}")
-            except Exception as e:
-                st.warning(f"Evaluation metrics unavailable: {str(e)}")
+        # Fixed input area at the top
+        with st.container():
+            st.markdown("<div class='input-container'>", unsafe_allow_html=True)
+            user_input = st.text_area("Ask a clinical question:", st.session_state.user_input, height=100, key="question_input")
+            col1, col2 = st.columns([1, 5])
+            with col1:
+                submit_button = st.button("Submit")
+            st.markdown("</div>", unsafe_allow_html=True)
         
-        # Query input
-        user_input = st.text_area("Ask a clinical question:", height=100)
-        col1, col2 = st.columns([1, 5])
-        with col1:
-            submit_button = st.button("Submit")
+        # Create a container for chat history
+        chat_container = st.container()
         
         # Process query
         if submit_button and user_input:
@@ -439,10 +427,40 @@ def main():
                         # Add to chat history
                         st.session_state.chat_history.append((user_input, response))
                         
+                        # Clear the input field
+                        st.session_state.user_input = ""
+                        
                         # Rerun to update UI
                         st.rerun()
                     except Exception as e:
                         st.error(f"Error processing query: {str(e)}")
+        
+        # Display chat history in the container
+        with chat_container:
+            for i, (query, response) in enumerate(st.session_state.chat_history):
+                st.markdown(f"<div class='chat-message chat-message-user'><b>🧑‍⚕️ You:</b> {query}</div>", unsafe_allow_html=True)
+                
+                st.markdown(f"<div class='chat-message chat-message-assistant'><b>🩺 DiReCT:</b> {response['answer']}</div>", unsafe_allow_html=True)
+                
+                with st.expander("View Sources"):
+                    for doc in response["context"]:
+                        st.markdown(f"<div class='source-box'>"
+                                  f"<b>Source:</b> {Path(doc.metadata['source']).stem}<br>"
+                                  f"<b>Type:</b> {doc.metadata['type']}<br>"
+                                  f"<b>Content:</b> {doc.page_content[:300]}...</div>", 
+                                  unsafe_allow_html=True)
+                
+                # Show evaluation metrics if available
+                try:
+                    eval_scores = evaluate_rag_response(response, embeddings)
+                    with st.expander("View Evaluation Metrics"):
+                        col1, col2 = st.columns(2)
+                        with col1:
+                            st.metric("Hit Rate (Top-3)", f"{eval_scores['hit_rate']:.2f}")
+                        with col2:
+                            st.metric("Faithfulness", f"{eval_scores['faithfulness']:.2f}")
+                except Exception as e:
+                    st.warning(f"Evaluation metrics unavailable: {str(e)}")
     
     # About page
     elif st.session_state.page == 'about':