Update app.py

app.py

@@ -0,0 +1,97 @@
+import gradio as gr
+import os
+import ezdxf
+import groq
+from ezdxf.entities import Insert, Line, Text, LWPolyline
+
+# Initialize Groq client
+client = groq.Client(api_key=os.getenv("GROQ_API_KEY"))
+
+# Parse the DXF file and extract basic entities
+def analyze_autocad_project(file_path):
+    try:
+        doc = ezdxf.readfile(file_path)
+        msp = doc.modelspace()
+
+        components = []
+        connections = []
+
+        for entity in msp:
+            if isinstance(entity, Insert):
+                name = entity.dxf.name
+                location = entity.dxf.insert
+                components.append({"type": "BLOCK", "name": name, "location": (location.x, location.y)})
+            elif isinstance(entity, Line):
+                start = (entity.dxf.start.x, entity.dxf.start.y)
+                end = (entity.dxf.end.x, entity.dxf.end.y)
+                connections.append({"type": "LINE", "start": start, "end": end})
+            elif isinstance(entity, LWPolyline):
+                points = entity.get_points("xy")
+                connections.append({"type": "POLYLINE", "points": points})
+            elif isinstance(entity, Text):
+                components.append({"type": "TEXT", "content": entity.dxf.text, "location": (entity.dxf.insert.x, entity.dxf.insert.y)})
+
+        summary = f"🔍 Found {len(components)} components and {len(connections)} connections."
+        return summary, {"components": components, "connections": connections}
+    except Exception as e:
+        return f"❌ Error: {str(e)}", None
+
+# RAG + Groq: Ask questions about specific components
+def ask_groq_question(context_dict, query):
+    try:
+        # Provide summarized context to Groq
+        component_text = "\n".join(
+            [f"{comp['type']} - {comp.get('name', comp.get('content', ''))} at {comp['location']}" for comp in context_dict["components"]]
+        )
+        connection_text = "\n".join(
+            [f"{conn['type']} from {conn.get('start', '')} to {conn.get('end', '')}" for conn in context_dict["connections"]]
+        )
+        context = f"COMPONENTS:\n{component_text}\n\nCONNECTIONS:\n{connection_text}\n\nQUESTION:\n{query}"
+
+        response = client.chat.completions.create(
+            model="llama3-8b-8192",
+            messages=[{"role": "user", "content": context}]
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Groq Error: {str(e)}"
+
+# Global store for RAG context
+rag_context = {}
+
+# Gradio Interface
+def upload_and_analyze(file):
+    summary, context = analyze_autocad_project(file.name)
+    if context:
+        global rag_context
+        rag_context = context
+    return summary
+
+def query_component(user_query):
+    if not rag_context:
+        return "⚠️ Please upload and analyze a DXF file first."
+    return ask_groq_question(rag_context, user_query)
+
+with gr.Blocks() as app:
+    gr.Markdown("# 🔌 AutoCAD Electrical Project Analyzer with Groq RAG")
+    gr.Markdown("""
+    Upload your AutoCAD DXF file to analyze the electrical project layout. Ask questions about specific components, connections, or trace paths.
+    """)
+
+    with gr.Row():
+        file_input = gr.File(label="📁 Upload AutoCAD DXF Project")
+        upload_btn = gr.Button("📊 Analyze Project")
+    
+    analysis_result = gr.Textbox(label="📝 Analysis Summary")
+
+    upload_btn.click(fn=upload_and_analyze, inputs=file_input, outputs=analysis_result)
+
+    gr.Markdown("### 💬 Ask about Component Back-tracing or Details")
+    user_query = gr.Textbox(label="🔍 Enter your question (e.g., trace component R1)")
+    answer_box = gr.Textbox(label="🧠 Groq RAG Answer")
+
+    ask_btn = gr.Button("🧠 Ask Groq")
+    ask_btn.click(fn=query_component, inputs=user_query, outputs=answer_box)
+
+if __name__ == "__main__":
+    app.launch()