Update app.py

app.py

@@ -1,14 +1,12 @@
-import gradio as gr
 import os
+import gradio as gr
 import ezdxf
+import tempfile
+import json
 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):
+# Function to parse the DXF and extract simple components and connections
+def analyze_dxf(file_path):
     try:
         doc = ezdxf.readfile(file_path)
         msp = doc.modelspace()
@@ -17,29 +15,35 @@ def analyze_autocad_project(file_path):
         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}
+            if entity.dxftype() == "INSERT":
+                components.append({
+                    "type": "BlockReference",
+                    "name": entity.dxf.name,
+                    "location": f"({entity.dxf.insert[0]}, {entity.dxf.insert[1]})"
+                })
+            elif entity.dxftype() == "TEXT":
+                components.append({
+                    "type": "Text",
+                    "content": entity.dxf.text,
+                    "location": f"({entity.dxf.insert[0]}, {entity.dxf.insert[1]})"
+                })
+            elif entity.dxftype() == "LINE":
+                connections.append({
+                    "type": "Line",
+                    "start": f"({entity.dxf.start[0]}, {entity.dxf.start[1]})",
+                    "end": f"({entity.dxf.end[0]}, {entity.dxf.end[1]})"
+                })
+
+        return components, connections
     except Exception as e:
-        return f"❌ Error: {str(e)}", None
+        return [], []
+
+# Function to ask Groq to analyze components
 
-# RAG + Groq: Ask questions about specific components
 def ask_groq_question(context_dict, query):
     try:
-        # Provide summarized context to Groq
+        client = groq.Client(api_key=os.getenv("GROQ_API_KEY"))
+
         component_text = "\n".join(
             [f"{comp['type']} - {comp.get('name', comp.get('content', ''))} at {comp['location']}" for comp in context_dict["components"]]
         )
@@ -56,42 +60,45 @@ def ask_groq_question(context_dict, query):
     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.
-    """)
+# Gradio interface
+def process_file(file):
+    if file is None:
+        return "⚠️ Please upload a valid AutoCAD DXF file."
+    components, connections = analyze_dxf(file.name)
+    context = {
+        "components": components,
+        "connections": connections
+    }
+    # Store context in a temp file to retrieve later
+    temp_path = os.path.join(tempfile.gettempdir(), "context.json")
+    with open(temp_path, "w") as f:
+        json.dump(context, f)
+    return "✅ File processed and understood. Now you can ask questions."
+
+def query_groq(user_query):
+    temp_path = os.path.join(tempfile.gettempdir(), "context.json")
+    if not os.path.exists(temp_path):
+        return "⚠️ No DXF context available. Please upload a file first."
+    with open(temp_path, "r") as f:
+        context = json.load(f)
+    return ask_groq_question(context, user_query)
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🧠 RAG-based AutoCAD Analyzer with Groq")
 
     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")
+        file_input = gr.File(label="📁 Upload AutoCAD DXF File", file_types=['.dxf'])
+        file_output = gr.Textbox(label="Status", interactive=False)
 
-    upload_btn.click(fn=upload_and_analyze, inputs=file_input, outputs=analysis_result)
+    process_btn = gr.Button("🔍 Analyze DXF")
+    process_btn.click(process_file, inputs=[file_input], outputs=[file_output])
 
-    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")
+    with gr.Row():
+        user_query = gr.Textbox(label="🔎 Ask a Question about the Circuit")
+        response = gr.Textbox(label="💬 Groq Answer")
 
-    ask_btn = gr.Button("🧠 Ask Groq")
-    ask_btn.click(fn=query_component, inputs=user_query, outputs=answer_box)
+    ask_btn = gr.Button("🤖 Ask Groq")
+    ask_btn.click(query_groq, inputs=[user_query], outputs=[response])
 
 if __name__ == "__main__":
-    app.launch()
+    demo.launch()