Update app.py

app.py

@@ -161,9 +161,11 @@ class StreamlitDocProcessor:
                 for file in sorted(st.session_state.processed_files['pdf']):
                     st.text(f"📄 {file}")
 
+    # Modify the qa_page method in the StreamlitDocProcessor class
+
     def qa_page(self):
         st.header("Query Documents")
-
+    
         try:
             # Refresh available files
             st.session_state.processed_files = self.get_processed_files()
@@ -171,45 +173,100 @@ class StreamlitDocProcessor:
             if not any(st.session_state.processed_files.values()):
                 st.warning("No processed files available. Please upload and process some files first.")
                 return
-
+    
             # Create combined list of files with icons
             all_files = []
             for file in st.session_state.processed_files.get('xml', []):
                 all_files.append(f"📱 {file}")
             for file in st.session_state.processed_files.get('pdf', []):
                 all_files.append(f"📄 {file}")
-
+    
             if not all_files:
                 st.warning("No processed files available. Please upload and process some files first.")
                 return
-
+    
             # File selection
             selected_files = st.multiselect(
                 "Select files to search through",
                 sorted(all_files),
                 default=all_files
             )
-
+    
             # Remove icons from selected files
             selected_files = [f.split(' ', 1)[1] for f in selected_files]
-
+    
             if not selected_files:
                 st.warning("Please select at least one file to search through.")
                 return
 
-            # Question input
-            question = st.text_input("Enter your question:")
+        # Question input
+        question = st.text_input("Enter your question:")
+        
+        if question:
+            col1, col2, col3 = st.columns(3)
             
-            if st.button("Ask Question") and question:
-                try:
-                    with st.spinner("Searching for answer..."):
-                        answer = st.session_state.processor.ask_question_selective(
-                            question,
-                            selected_files
-                        )
-                        st.write("Answer:", answer)
-                except Exception as e:
-                    st.error(f"Error getting answer: {str(e)}")
+            with col1:
+                if st.button("Quick Answer"):
+                    try:
+                        with st.spinner("Getting quick answer..."):
+                            answer = st.session_state.processor.ask_question_selective(
+                                question,
+                                selected_files
+                            )
+                            st.write("Answer:", answer)
+                    except Exception as e:
+                        st.error(f"Error getting answer: {str(e)}")
+
+            with col2:
+                if st.button("Detailed Answer"):
+                    try:
+                        with st.spinner("Getting detailed answer..."):
+                            result = st.session_state.processor.get_detailed_context(
+                                question,
+                                selected_files
+                            )
+                            if result['success']:
+                                st.write("### Relevant Information")
+                                for item in result['results']:
+                                    with st.expander(f"Source: {item['metadata']['source_file']} ({item['metadata']['content_type'].upper()})"):
+                                        st.write(f"Relevance Score: {item['relevance_score']:.2f}")
+                                        if item['metadata']['content_type'] == 'xml':
+                                            st.write(f"XML Path: {item['source_info']['path']}")
+                                        st.write("Content:", item['content'])
+                            else:
+                                st.error(result['error'])
+                    except Exception as e:
+                        st.error(f"Error getting detailed answer: {str(e)}")
+
+            with col3:
+                if st.button("Complete Analysis"):
+                    try:
+                        with st.spinner("Performing complete analysis..."):
+                            result = st.session_state.processor.get_summary_and_details(
+                                question,
+                                selected_files
+                            )
+                            if result['success']:
+                                st.write("### Summary")
+                                st.write(result['summary'])
+                                
+                                st.write("### Detailed Information")
+                                for item in result['details']:
+                                    with st.expander(f"Source: {item['metadata']['source_file']} ({item['metadata']['content_type'].upper()})"):
+                                        st.write(f"Relevance Score: {item['relevance_score']:.2f}")
+                                        if item['metadata']['content_type'] == 'xml':
+                                            st.write(f"XML Path: {item['source_info']['path']}")
+                                            if 'parent_info' in item:
+                                                st.write("Parent Element:", item['parent_info']['content'])
+                                            if 'children_info' in item:
+                                                st.write("Related Elements:")
+                                                for child in item['children_info']:
+                                                    st.write(f"- {child['content']}")
+                                        st.write("Content:", item['content'])
+                            else:
+                                st.error(result['error'])
+                    except Exception as e:
+                        st.error(f"Error getting complete analysis: {str(e)}")
                     
         except Exception as e:
             st.error(f"Error in Q&A interface: {str(e)}")