Create app.py

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+import pandas as pd
+import plotly.graph_objects as go
+import numpy as np
+
+def plot_zip_code_correlation(zip_codes_str):
+    # Convert input string to list of zip codes
+    zip_codes = [z.strip() for z in zip_codes_str.split(",")]
+
+    # Read the CSV file
+    df = pd.read_csv('https://files.zillowstatic.com/research/public_csvs/zhvi/Zip_zhvi_uc_sfrcondo_tier_0.33_0.67_sm_sa_month.csv')
+
+    # Filter for the given ZIP codes
+    df = df[df['RegionName'].astype(str).isin(zip_codes)]
+
+    if df.empty:
+        raise ValueError("No data found for the provided ZIP codes.")
+
+    # Extract columns that are valid date strings only
+    date_columns = []
+    for col in df.columns[7:]:
+        try:
+            pd.to_datetime(col)
+            date_columns.append(col)
+        except:
+            continue
+
+    # Initialize a DataFrame to hold price data for correlation calculation
+    price_matrix = []
+    zip_list = []
+
+    for zip_code in zip_codes:
+        df_zip = df[df['RegionName'].astype(str) == zip_code]
+        if not df_zip.empty:
+            prices = df_zip.loc[:, date_columns].values.flatten()
+            if not np.isnan(prices).all():
+                price_matrix.append(prices)
+                zip_list.append(zip_code)
+
+    # Check if there is enough data
+    if len(price_matrix) < 2:
+        raise ValueError("Not enough data for correlation calculation.")
+
+    price_matrix_df = pd.DataFrame(price_matrix, index=zip_list, columns=date_columns)
+    price_matrix_df = price_matrix_df.T.dropna()
+
+    # Calculate correlation
+    corr_matrix = price_matrix_df.corr()
+
+    # Prepare data for 3D plot
+    z_data = corr_matrix.values
+    x_data, y_data = np.meshgrid(zip_list, zip_list)
+
+    # Create the 3D surface plot
+    fig = go.Figure(data=[go.Surface(z=z_data, x=x_data, y=y_data)])
+
+    fig.update_layout(
+        title='3D Correlation Matrix of Housing Prices for Selected ZIP Codes',
+        scene=dict(
+            xaxis_title='ZIP Code',
+            yaxis_title='ZIP Code',
+            zaxis_title='Correlation',
+        ),
+        autosize=True
+    )
+
+    return fig
+
+iface = gr.Interface(
+    fn=plot_zip_code_correlation,
+    inputs=gr.Textbox(label="Enter comma-separated ZIP codes (e.g., 07001, 07002, 07003)"),
+    outputs=gr.Plot(),
+    title="ZIP Code 3D Correlation Matrix"
+)
+
+iface.launch(debug=True)