Update app.py

app.py

@@ -4,61 +4,98 @@ from data_processor import DataProcessor
 from visualizer import Visualizer
 from analyzer import Analyzer
 from openai_agent import OpenAIAgent
+from time_series_analyzer import TimeSeriesAnalyzer
+from machine_learning import MachineLearning
+from text_analyzer import TextAnalyzer
+from geospatial_analyzer import GeospatialAnalyzer
+from utils import load_data, save_data
 
-st.set_page_config(page_title="AI-Powered Data Viz", layout="wide")
+st.set_page_config(page_title="AI-Powered Data Analysis", layout="wide")
 
-st.title("AI-Powered Data Visualization")
+st.title("AI-Powered Data Analysis and Visualization")
 
 # File uploader
-uploaded_file = st.file_uploader("Choose a CSV file", type="csv")
+uploaded_file = st.sidebar.file_uploader("Choose a CSV file", type="csv")
 
 if uploaded_file is not None:
-    # Read the CSV file
-    df = pd.read_csv(uploaded_file)
+    df = load_data(uploaded_file)
     
-    # Initialize our components
+    # Initialize components
     data_processor = DataProcessor(df)
     visualizer = Visualizer()
     analyzer = Analyzer()
     openai_agent = OpenAIAgent()
+    time_series_analyzer = TimeSeriesAnalyzer()
+    ml = MachineLearning()
+    text_analyzer = TextAnalyzer()
+    geo_analyzer = GeospatialAnalyzer()
 
-    # Data Cleaning and Preprocessing
-    st.header("Data Cleaning and Preprocessing")
-    st.write("Original data shape:", df.shape)
-    st.write("Columns with missing values:", data_processor.get_columns_with_missing_values())
-    
-    if st.button("Clean Data"):
+    # Sidebar for feature selection
+    feature = st.sidebar.radio(
+        "Select a feature",
+        ["Data Overview", "Data Cleaning", "Data Visualization", "Data Analysis", 
+         "AI Insights", "Time Series Analysis", "Machine Learning", 
+         "Text Analysis", "Geospatial Analysis", "Custom AI Tasks"]
+    )
+
+    if feature == "Data Overview":
+        st.header("Data Overview")
+        st.write("Dataset shape:", df.shape)
+        st.write("Columns:", df.columns.tolist())
+        st.write("Sample data:")
+        st.write(df.head())
+        
+        if st.checkbox("Show data types"):
+            st.write(df.dtypes)
+        
+        if st.checkbox("Show summary statistics"):
+            st.write(df.describe())
+
+    elif feature == "Data Cleaning":
+        st.header("Data Cleaning and Preprocessing")
         df_cleaned = data_processor.clean_data()
         st.write("Cleaned data shape:", df_cleaned.shape)
         st.write(df_cleaned.head())
+        
+        if st.button("Save Cleaned Data"):
+            save_data(df_cleaned, "cleaned_data.csv")
+            st.success("Cleaned data saved successfully!")
 
-    # Data Visualization
-    st.header("Data Visualization")
-    columns = df.columns.tolist()
-    x_axis = st.selectbox("Select X-axis", columns)
-    y_axis = st.selectbox("Select Y-axis", columns)
-    
-    chart_type = st.radio("Select chart type", ["Scatter", "Line", "Bar"])
-    
-    if st.button("Generate Visualization"):
-        fig = visualizer.create_plot(df, x_axis, y_axis, chart_type)
-        st.plotly_chart(fig)
+    elif feature == "Data Visualization":
+        st.header("Data Visualization")
+        visualizer.create_visualizations(df)
 
-    # Data Analysis
-    st.header("Data Analysis")
-    analysis_prompt = st.text_input("Enter your analysis question:")
-    
-    if st.button("Analyze"):
-        analysis_result = analyzer.analyze_data(df, analysis_prompt)
-        st.write(analysis_result)
+    elif feature == "Data Analysis":
+        st.header("Data Analysis")
+        analyzer.perform_analysis(df)
 
-    # OpenAI Integration
-    st.header("AI-Powered Insights")
-    ai_prompt = st.text_area("Ask the AI for insights:")
-    
-    if st.button("Get AI Insights"):
-        ai_response = openai_agent.get_insights(df, ai_prompt)
-        st.write(ai_response)
+    elif feature == "AI Insights":
+        st.header("AI-Powered Insights")
+        openai_agent.generate_insights(df)
+
+    elif feature == "Time Series Analysis":
+        st.header("Time Series Analysis")
+        time_series_analyzer.analyze(df)
+
+    elif feature == "Machine Learning":
+        st.header("Machine Learning")
+        ml.perform_ml_tasks(df)
+
+    elif feature == "Text Analysis":
+        st.header("Text Analysis")
+        text_analyzer.analyze_text(df)
+
+    elif feature == "Geospatial Analysis":
+        st.header("Geospatial Analysis")
+        geo_analyzer.analyze_geospatial_data(df)
+
+    elif feature == "Custom AI Tasks":
+        st.header("Custom AI Tasks")
+        openai_agent.custom_ai_task(df)
 
 else:
-    st.write("Please upload a CSV file to get started.")
+    st.write("Please upload a CSV file to get started.")
+
+# Add a footer
+st.sidebar.markdown("---")
+st.sidebar.info("Developed by Your Company Name")