Update app.py

app.py

@@ -1,92 +1,64 @@
 import streamlit as st
-import sys
-import subprocess
+import matplotlib.pyplot as plt
+import numpy as np
 
-# Install compatible version of PyPDF2
-subprocess.check_call([sys.executable, "-m", "pip", "install", "PyPDF2==2.12.1"])
+def draw_circle(ax, center, radius, color='black', fill=False, lw=1.5):
+    circle = plt.Circle(center, radius, clip_on=False, edgecolor=color, facecolor='none' if not fill else color, lw=lw, fill=fill)
+    ax.add_artist(circle)
 
-import camelot
-import fitz  # PyMuPDF
-import pandas as pd
-import os
-import tempfile
+def draw_line(ax, start, end, color='black', lw=1.5):
+    ax.plot([start[0], end[0]], [start[1], end[1]], color=color, lw=lw)
 
-# Set the title of the Streamlit app
-st.title("PDF Table Extractor")
+def setup_plot(xlim=(-1.5, 1.5), ylim=(-1.5, 1.5), aspect='equal'):
+    fig, ax = plt.subplots(figsize=(6, 6))
+    ax.set_xlim(xlim)
+    ax.set_ylim(ylim)
+    ax.set_aspect(aspect)
+    ax.axis('off')
+    return fig, ax
 
-# Instructions
-st.write("Upload a PDF file containing tables, and this app will extract the tables for you.")
+def draw_shree_yantra(ax, radius=1, sides=12, line_skips=[2,5,7,10], circle_fill=False, line_color='black', circle_color='black'):
+    # Step 1: Draw outer circle
+    draw_circle(ax, (0, 0), radius, color=circle_color, fill=circle_fill, lw=2)
 
-# File uploader widget
-uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
+    # Step 2: Draw polygon (e.g., dodecagon for 12 sides)
+    points = [np.array([np.cos(theta), np.sin(theta)]) for theta in np.linspace(0, 2*np.pi, sides, endpoint=False)]
+    polygon = plt.Polygon(points, closed=True, fill=False, edgecolor=circle_color, lw=2)
+    ax.add_artist(polygon)
 
-if uploaded_file is not None:
-    try:
-        # Load the uploaded PDF using PyMuPDF
-        pdf_document = fitz.open(stream=uploaded_file.read(), filetype="pdf")
-        
-        # Show the number of pages in the PDF
-        num_pages = pdf_document.page_count
-        st.write(f"The uploaded PDF has {num_pages} pages.")
-        
-        # Let the user select pages to extract tables from
-        page_range = st.text_input("Enter page range (e.g., 1-3,5,7-9) or 'all' for all pages:", "all")
-        
-        # Camelot options
-        flavor = st.selectbox("Select table flavor:", ["lattice", "stream"])
-        
-        # Extract tables using Camelot
-        if st.button("Extract Tables"):
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
-                temp_file.write(uploaded_file.getvalue())
-                temp_file_path = temp_file.name
-            
-            try:
-                with st.spinner("Extracting tables..."):
-                    if page_range.lower() == 'all':
-                        pages = str(list(range(1, num_pages + 1)))[1:-1]
-                    else:
-                        pages = page_range
-                    
-                    tables = camelot.read_pdf(temp_file_path, pages=pages, flavor=flavor)
-                
-                if len(tables) > 0:
-                    st.write(f"Found {len(tables)} table(s).")
-                    
-                    # Loop through all the extracted tables
-                    for i, table in enumerate(tables):
-                        st.write(f"Table {i+1} (Page {table.parsing_report['page']}):")
-                        
-                        # Convert the table to a Pandas DataFrame
-                        df = table.df
-                        
-                        # Display the extracted table
-                        st.dataframe(df)
-                        
-                        # Download buttons
-                        col1, col2 = st.columns(2)
-                        with col1:
-                            csv = df.to_csv(index=False).encode('utf-8')
-                            st.download_button(
-                                label=f"Download as CSV",
-                                data=csv,
-                                file_name=f"table_{i+1}_page_{table.parsing_report['page']}.csv",
-                                mime="text/csv"
-                            )
-                        with col2:
-                            excel_file = df.to_excel(index=False, engine="xlsxwriter")
-                            st.download_button(
-                                label=f"Download as Excel",
-                                data=excel_file,
-                                file_name=f"table_{i+1}_page_{table.parsing_report['page']}.xlsx",
-                                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-                            )
-                else:
-                    st.write("No tables found in the selected pages.")
-            
-            finally:
-                # Clean up the temporary file
-                os.unlink(temp_file_path)
-    
-    except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
+    # Step 3: Draw connecting lines based on skips
+    for i, p1 in enumerate(points):
+        for j, p2 in enumerate(points):
+            if i != j and abs(i - j) in line_skips:
+                draw_line(ax, p1 * radius, p2 * radius, color=line_color, lw=1.5)
+
+    # Additional steps can be implemented here following the traditional Shree Yantra construction
+    # For simplicity, this example draws a basic pattern
+
+def main():
+    st.title("Shree Yantra Diagram Generator")
+
+    st.sidebar.header("Customize Your Shree Yantra")
+
+    # Sidebar controls
+    radius = st.sidebar.slider("Outer Circle Radius", 0.5, 2.0, 1.0, 0.1)
+    sides = st.sidebar.slider("Number of Sides (Polygon)", 6, 20, 12, 1)
+    line_skips = st.sidebar.multiselect("Line Skips", options=list(range(1, sides//2 +1)), default=[2,5,7,10])
+    line_color = st.sidebar.color_picker("Line Color", "#000000")
+    circle_color = st.sidebar.color_picker("Circle & Polygon Color", "#000000")
+    circle_fill = st.sidebar.checkbox("Fill Outer Circle", value=False)
+
+    st.write("### Shree Yantra Diagram")
+
+    fig, ax = setup_plot()
+
+    if not line_skips:
+        st.warning("Please select at least one line skip to draw connecting lines.")
+    else:
+        draw_shree_yantra(ax, radius=radius, sides=sides, line_skips=line_skips, 
+                         circle_fill=circle_fill, line_color=line_color, circle_color=circle_color)
+
+    st.pyplot(fig)
+
+if __name__ == "__main__":
+    main()