Create app.py

app.py

@@ -0,0 +1,92 @@
+import os
+from transformers import pipeline
+import streamlit as st
+from PyPDF2 import PdfReader
+from groq import Groq
+
+# Initialize Groq API client
+client = Groq(api_key="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")
+
+# Example CAD metadata
+cad_data = [
+    {"id": 1, "name": "Gear", "description": "A gear with specified teeth and diameter."},
+    {"id": 2, "name": "Bracket", "description": "A structural bracket for support."}
+]
+
+# Function to retrieve CAD templates using Groq API
+def retrieve_cad_template_with_groq(query):
+    messages = [
+        {
+            "role": "user",
+            "content": f"Find a CAD template matching this query: {query}",
+        }
+    ]
+    chat_completion = client.chat.completions.create(
+        messages=messages, model="llama-3.3-70b-versatile"
+    )
+    # Extract the response
+    response = chat_completion.choices[0].message.content
+    return response
+
+# Function to process PDF input
+def process_pdf(file_path):
+    reader = PdfReader(file_path)
+    text = ""
+    for page in reader.pages:
+        text += page.extract_text()
+    return text[:500]  # Limit to 500 characters for API processing
+
+# Function to generate OpenSCAD script for gear
+def generate_gear(teeth, diameter):
+    scad_script = f"""
+    module gear() {{
+        difference() {{
+            circle(r={diameter / 2}, $fn=teeth);
+            translate([0, 0, -1]) circle(r={diameter / 3}, $fn=teeth);
+        }}
+    }}
+    gear();
+    """
+    file_path = "generated_model.scad"
+    with open(file_path, "w") as file:
+        file.write(scad_script)
+    return file_path
+
+# Streamlit UI
+st.title("Quick CAD Model Generator")
+st.sidebar.title("Input Options")
+input_option = st.sidebar.radio("Select Input Type", ["Text Description", "Upload PDF", "Google Drive Link"])
+
+# Process input
+query = ""
+if input_option == "Text Description":
+    query = st.text_input("Describe the part you need (e.g., 'A gear with 20 teeth').")
+elif input_option == "Upload PDF":
+    pdf_file = st.file_uploader("Upload a PDF with a CAD drawing", type=["pdf"])
+    if pdf_file:
+        query = process_pdf(pdf_file)
+        st.write("Extracted Text:", query)
+elif input_option == "Google Drive Link":
+    drive_link = st.text_input("Enter Google Drive Link for the CAD file")
+    if drive_link:
+        st.write(f"Processing file from: {drive_link}")
+        query = f"File from Google Drive: {drive_link}"
+
+# Template suggestion using Groq API
+if st.button("Find Template") and query:
+    try:
+        suggested_template = retrieve_cad_template_with_groq(query)
+        st.success(f"Suggested Template: {suggested_template}")
+    except Exception as e:
+        st.error(f"Error using Groq API: {e}")
+
+# Generate model
+part_type = st.selectbox("Select Part Type", ["Gear", "Bracket"])
+teeth = st.number_input("Number of Teeth (for Gears)", min_value=10, max_value=100, step=1, value=20)
+diameter = st.number_input("Diameter (mm)", min_value=10, max_value=200, step=1, value=50)
+
+if st.button("Generate CAD Model"):
+    if part_type == "Gear":
+        file_path = generate_gear(teeth=teeth, diameter=diameter)
+        st.success(f"CAD Model Generated: {file_path}")
+        st.download_button("Download Model", open(file_path, "rb").read(), file_name="generated_model.scad")