Update app.py

app.py

@@ -139,31 +139,30 @@ def main():
     # User input
     user_query = st.text_area("Enter your query:", height=150)
     submit_button = st.button("Analyze Query")
+    # Create tabs
+    tab1, tab2, tab3 = st.tabs(["Overview", "Analysis", "Sentiment"])
 
-    if submit_button and user_query:
-        # Fetching response from Meta AI
-        response = fetch_response(user_query)
-        msg = response.get('message', 'No response message.')    
-        # Write response
-        st.write(msg)
-
-        # Run sentiment analysis
-        df_sentiment = sentiment_analysis(msg)
-
-        # Fetch advanced analysis
-        advanced_response = fetch_advanced_analysis(user_query, msg)
-        advanced_msg = advanced_response.get('message', 'No advanced analysis available.')
-
-        # Parse the advanced analysis response
-        analysis_data = parse_analysis(advanced_msg)
+    # Tab 1: Overview - Showing the initial response and sources
+    with tab1:
+        user_query = st.text_area("Enter your query:", height=150)
+        submit_button = st.button("Analyze Query")
         
-        if "error" in analysis_data:
-            st.error("Error in analysis: " + analysis_data["error"])
-            if "details" in analysis_data:
-                st.error("Details: " + analysis_data["details"])
-        else:
-            # Display parsed data in a collapsible section
-            with st.expander("Show Advanced Analysis"):
+        if submit_button and user_query:
+            response = fetch_response(user_query)
+            msg = response.get('message', 'No response message.')
+            st.write(msg)
+            
+            with st.expander("Show Sources"):
+                display_sources(response.get('sources', []))
+
+    # Tab 2: Analysis - Showing the result of the advanced analysis
+    with tab2:
+        if submit_button and user_query:
+            advanced_response = fetch_advanced_analysis(user_query, msg)
+            advanced_msg = advanced_response.get('message', 'No advanced analysis available.')
+            analysis_data = parse_analysis(advanced_msg)
+            
+            if "error" not in analysis_data:
                 st.write("### User Intent")
                 st.write(analysis_data['user_intent'])
                 st.write("### Follow-up Questions")
@@ -173,10 +172,10 @@ def main():
                 for entity_type, entities in analysis_data['entities'].items():
                     st.write(f"**{entity_type.capitalize()}**: {', '.join(entities)}")
 
-
-        # Display the sentiment in a collapsible section
-        with st.expander("Show Sentiment"):
-            # Display negative sentence locations
+    # Tab 3: Sentiment - Displaying sentiment analysis of the response
+    with tab3:
+        if submit_button and user_query:
+            df_sentiment = sentiment_analysis(msg)
             fig = px.scatter(df_sentiment, y='dominant_sentiment', color='dominant_sentiment', size='confidence', 
                              hover_data=['content'], 
                              color_discrete_map={"neg": "firebrick", "neu": "navajowhite", "pos": "darkgreen"},
@@ -184,11 +183,6 @@ def main():
                              title='Sentiment Analysis of the Response')
             fig.update_layout(width=800, height=300)
             st.plotly_chart(fig)
-        
-        # Display the AI response in a collapsible section
-        with st.expander("Show Sources"):
-            # Display sources with clickable links in a collapsible section
-            display_sources(response.get('sources', []))
-                    
+
 if __name__ == "__main__":
-    main()
+    main()