Add .gitignore and remove incomplete file
This commit is contained in:
JC Beasley
+15
@@ -0,0 +1,15 @@
|
||||
# Python
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
venv/
|
||||
|
||||
# Uploads
|
||||
uploads/
|
||||
|
||||
# IDE
|
||||
.vscode/
|
||||
.idea/
|
||||
|
||||
# OS
|
||||
.DS_Store
|
||||
|
||||
-663
@@ -1,663 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
IT Site Survey AI Application
|
||||
Complete implementation with all requested features:
|
||||
- Photo uploads for AI assessment
|
||||
- Model selection (Ollama vision models)
|
||||
- Export results (PDF, Text, Email)
|
||||
- Delete surveys (individual + Clear All)
|
||||
- Chat interface for AI interaction
|
||||
- Load Sample Data for testing
|
||||
- Comprehensive 35-question survey template
|
||||
"""
|
||||
|
||||
from flask import Flask, request, jsonify, send_file
|
||||
from flask_cors import CORS
|
||||
import os
|
||||
import json
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
import base64
|
||||
import requests
|
||||
from werkzeug.utils import secure_filename
|
||||
|
||||
app = Flask(__name__)
|
||||
CORS(app, resources={
|
||||
r"/api/*": {
|
||||
"origins": ["http://192.168.50.11:3003", "http://localhost:3003"],
|
||||
"methods": ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
|
||||
"allow_headers": ["Content-Type", "Authorization"]
|
||||
}
|
||||
})
|
||||
|
||||
# Configuration
|
||||
UPLOAD_FOLDER = 'uploads'
|
||||
PHOTOS_FOLDER = 'uploads/photos'
|
||||
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
|
||||
OLLAMA_HOST = os.getenv('OLLAMA_HOST', 'http://192.168.19.25:11434')
|
||||
OLLAMA_URL = f"{OLLAMA_HOST}/api/chat"
|
||||
|
||||
# Create upload directories
|
||||
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
|
||||
os.makedirs(PHOTOS_FOLDER, exist_ok=True)
|
||||
|
||||
# In-memory storage
|
||||
surveys_db = {}
|
||||
responses_db = {}
|
||||
|
||||
# Survey Template with 35 comprehensive questions
|
||||
DEFAULT_SURVEY_TEMPLATE = {
|
||||
"title": "IT Infrastructure Site Survey",
|
||||
"description": "Comprehensive network infrastructure assessment",
|
||||
"questions": [
|
||||
# Site Topology & Layout (1-4)
|
||||
{
|
||||
"id": "q1",
|
||||
"category": "Site Topology & Layout",
|
||||
"question": "What is the total square footage of the site?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 3500"
|
||||
},
|
||||
{
|
||||
"id": "q2",
|
||||
"category": "Site Topology & Layout",
|
||||
"question": "How many floors or buildings need network coverage?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 2"
|
||||
},
|
||||
{
|
||||
"id": "q3",
|
||||
"category": "Site Topology & Layout",
|
||||
"question": "Describe the site layout (open plan, offices, warehouse, etc.)",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe the layout..."
|
||||
},
|
||||
{
|
||||
"id": "q4",
|
||||
"category": "Site Topology & Layout",
|
||||
"question": "Are there any physical barriers that may affect WiFi coverage?",
|
||||
"type": "multiselect",
|
||||
"options": ["Concrete walls", "Metal shelving", "Elevator shafts", "Thick drywall", "None"]
|
||||
},
|
||||
|
||||
# Network Closet & Equipment (5-8)
|
||||
{
|
||||
"id": "q5",
|
||||
"category": "Network Closet & Equipment",
|
||||
"question": "How many network closets or IDF rooms are on-site?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 2"
|
||||
},
|
||||
{
|
||||
"id": "q6",
|
||||
"category": "Network Closet & Equipment",
|
||||
"question": "Describe the current network closet setup",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe the setup..."
|
||||
},
|
||||
{
|
||||
"id": "q7",
|
||||
"category": "Network Closet & Equipment",
|
||||
"question": "What type of power and cooling is available in the network closets?",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe power and cooling..."
|
||||
},
|
||||
{
|
||||
"id": "q8",
|
||||
"category": "Network Closet & Equipment",
|
||||
"question": "Are there existing server racks or wall-mounted equipment?",
|
||||
"type": "select",
|
||||
"options": ["Yes, full racks", "Yes, partial racks", "Wall-mounted only", "No existing racks"]
|
||||
},
|
||||
|
||||
# Cabling Infrastructure (9-12)
|
||||
{
|
||||
"id": "q9",
|
||||
"category": "Cabling Infrastructure",
|
||||
"question": "What type of existing cabling is in place?",
|
||||
"type": "select",
|
||||
"options": ["Cat5e (100Mb-1GbE)", "Cat6 (1GbE-10GbE)", "Cat6a (10GbE)", "Fiber (Multi-mode)", "Fiber (Single-mode)", "No existing cabling", "Unknown"]
|
||||
},
|
||||
{
|
||||
"id": "q10",
|
||||
"category": "Cabling Infrastructure",
|
||||
"question": "How many network drops are currently installed?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 120"
|
||||
},
|
||||
{
|
||||
"id": "q11",
|
||||
"category": "Cabling Infrastructure",
|
||||
"question": "What is the estimated distance from the MDF to the farthest IDF or workstation?",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe distances..."
|
||||
},
|
||||
{
|
||||
"id": "q12",
|
||||
"category": "Cabling Infrastructure",
|
||||
"question": "Are there any special cabling requirements?",
|
||||
"type": "multiselect",
|
||||
"options": ["Plenum-rated cable required", "Outdoor-rated cable needed", "Direct burial cable needed", "Aerial cable needed", "Shielded cable required", "None"]
|
||||
},
|
||||
|
||||
# Network Equipment (13-15)
|
||||
{
|
||||
"id": "q13",
|
||||
"category": "Network Equipment",
|
||||
"question": "What network equipment is currently installed?",
|
||||
"type": "multiselect",
|
||||
"options": ["Router/Firewall", "Managed switches", "Unmanaged switches", "Wireless access points", "None/Unknown"]
|
||||
},
|
||||
{
|
||||
"id": "q14",
|
||||
"category": "Network Equipment",
|
||||
"question": "What type of switches are needed?",
|
||||
"type": "select",
|
||||
"options": ["Unmanaged (plug-and-play)", "Managed (Layer 2)", "Managed (Layer 3 with routing)", "PoE switches", "Industrial-grade switches"]
|
||||
},
|
||||
{
|
||||
"id": "q15",
|
||||
"category": "Network Equipment",
|
||||
"question": "How many wireless access points are needed?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 6"
|
||||
},
|
||||
|
||||
# Power Requirements (16-18)
|
||||
{
|
||||
"id": "q16",
|
||||
"category": "Power Requirements",
|
||||
"question": "What is the estimated total power consumption for all network equipment?",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe power requirements..."
|
||||
},
|
||||
{
|
||||
"id": "q17",
|
||||
"category": "Power Requirements",
|
||||
"question": "What PoE devices will be connected?",
|
||||
"type": "multiselect",
|
||||
"options": ["IP Phones", "Wireless APs", "IP Cameras", "IoT sensors", "None", "Not sure"]
|
||||
},
|
||||
{
|
||||
"id": "q18",
|
||||
"category": "Power Requirements",
|
||||
"question": "What UPS backup capacity is required?",
|
||||
"type": "select",
|
||||
"options": ["No backup needed", "Minimal backup (15-30 min)", "Standard backup (1-2 hours)", "Extended backup (4+ hours)", "Generator backup"]
|
||||
},
|
||||
|
||||
# User & Device Requirements (19-21)
|
||||
{
|
||||
"id": "q19",
|
||||
"category": "User & Device Requirements",
|
||||
"question": "How many employees or users will use the network?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 45"
|
||||
},
|
||||
{
|
||||
"id": "q20",
|
||||
"category": "User & Device Requirements",
|
||||
"question": "What is the estimated total number of devices?",
|
||||
"type": "number",
|
||||
"placeholder": "e.g., 85"
|
||||
},
|
||||
{
|
||||
"id": "q21",
|
||||
"category": "User & Device Requirements",
|
||||
"question": "What types of network activities will be performed?",
|
||||
"type": "multiselect",
|
||||
"options": ["Email/Web Browsing", "Video conferencing", "File sharing", "Cloud applications", "Remote desktop", "VoIP/Phone calls", "Streaming media"]
|
||||
},
|
||||
|
||||
# Security & Compliance (22-24)
|
||||
{
|
||||
"id": "q22",
|
||||
"category": "Security & Compliance",
|
||||
"question": "What compliance standards must be met?",
|
||||
"type": "multiselect",
|
||||
"options": ["HIPAA", "PCI-DSS", "SOC 2", "GDPR", "NIST", "None/General security"]
|
||||
},
|
||||
{
|
||||
"id": "q23",
|
||||
"category": "Security & Compliance",
|
||||
"question": "What network segmentation is required?",
|
||||
"type": "multiselect",
|
||||
"options": ["Corporate network", "Guest network", "IoT network", "Management network", "No segmentation needed"]
|
||||
},
|
||||
{
|
||||
"id": "q24",
|
||||
"category": "Security & Compliance",
|
||||
"question": "What security features are needed?",
|
||||
"type": "multiselect",
|
||||
"options": ["Firewall", "VPN access", "Intrusion detection", "Network access control (NAC)", "Content filtering", "None"]
|
||||
},
|
||||
|
||||
# Internet & WAN (25-27)
|
||||
{
|
||||
"id": "q25",
|
||||
"category": "Internet & WAN",
|
||||
"question": "What type of internet connectivity is available?",
|
||||
"type": "multiselect",
|
||||
"options": ["Fiber", "Cable/DSL", "5G/Cellular backup", "Dedicated leased line", "No existing connection"]
|
||||
},
|
||||
{
|
||||
"id": "q26",
|
||||
"category": "Internet & WAN",
|
||||
"question": "What is the required internet bandwidth?",
|
||||
"type": "textarea",
|
||||
"placeholder": "e.g., 500Mbps download / 250Mbps upload"
|
||||
},
|
||||
{
|
||||
"id": "q27",
|
||||
"category": "Internet & WAN",
|
||||
"question": "Are there multiple sites that need to be connected?",
|
||||
"type": "select",
|
||||
"options": ["No - Single site only", "Yes - 2-3 sites", "Yes - 4+ sites", "Not sure yet"]
|
||||
},
|
||||
|
||||
# Existing Infrastructure (28-30)
|
||||
{
|
||||
"id": "q28",
|
||||
"category": "Existing Infrastructure",
|
||||
"question": "Is there existing IT infrastructure to integrate with?",
|
||||
"type": "select",
|
||||
"options": ["Yes - Full upgrade needed", "Yes - Partial upgrade", "No - New installation", "Not sure"]
|
||||
},
|
||||
{
|
||||
"id": "q29",
|
||||
"category": "Existing Infrastructure",
|
||||
"question": "Describe any existing network issues or limitations",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe current issues..."
|
||||
},
|
||||
{
|
||||
"id": "q30",
|
||||
"category": "Existing Infrastructure",
|
||||
"question": "What performance issues have been observed?",
|
||||
"type": "multiselect",
|
||||
"options": ["Slow internet", "WiFi dead zones", "Network congestion", "Frequent disconnections", "No issues observed"]
|
||||
},
|
||||
|
||||
# Budget & Timeline (31-33)
|
||||
{
|
||||
"id": "q31",
|
||||
"category": "Budget & Timeline",
|
||||
"question": "What is the estimated budget for this project?",
|
||||
"type": "select",
|
||||
"options": ["Under $10,000", "$10,000 - $25,000", "$25,000 - $50,000", "$50,000 - $100,000", "$100,000+", "Not determined yet"]
|
||||
},
|
||||
{
|
||||
"id": "q32",
|
||||
"category": "Budget & Timeline",
|
||||
"question": "What is the desired timeline for completion?",
|
||||
"type": "select",
|
||||
"options": ["Within 1 month", "Within 3 months", "Within 6 months", "Within 1 year", "Flexible"]
|
||||
},
|
||||
{
|
||||
"id": "q33",
|
||||
"category": "Budget & Timeline",
|
||||
"question": "Who will be responsible for ongoing network management?",
|
||||
"type": "select",
|
||||
"options": ["Internal IT team", "Managed service provider (MSP)", "Hybrid approach", "Not determined yet"]
|
||||
},
|
||||
|
||||
# Additional Details (34-35)
|
||||
{
|
||||
"id": "q34",
|
||||
"category": "Additional Details",
|
||||
"question": "Upload any site photos, floor plans, or network diagrams",
|
||||
"type": "file",
|
||||
"placeholder": "Upload photos..."
|
||||
},
|
||||
{
|
||||
"id": "q35",
|
||||
"category": "Additional Details",
|
||||
"question": "Any additional requirements or special considerations?",
|
||||
"type": "textarea",
|
||||
"placeholder": "Describe any additional requirements..."
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
# Sample data for testing
|
||||
SAMPLE_SURVEY_DATA = {
|
||||
"name": "Sample Corporate Office Survey",
|
||||
"client_name": "Acme Corporation",
|
||||
"site_name": "Main Office - Downtown",
|
||||
"description": "Comprehensive network infrastructure assessment for downtown office location",
|
||||
"responses": {
|
||||
"q1": "3500",
|
||||
"q2": "2",
|
||||
"q3": "Single-story office building with open floor plan, 8 private offices, 2 conference rooms, break room, and reception area. Approximately 60 workstations in cubicle area.",
|
||||
"q4": ["Concrete walls"],
|
||||
"q5": "2",
|
||||
"q6": "MDF located in basement utility room. IDF on first floor in IT storage closet. Both have adequate ventilation but limited space.",
|
||||
"q7": "2x 20A circuits per closet, UPS backup planned for MDF",
|
||||
"q8": "Yes, partial racks",
|
||||
"q9": "Cat5e (100Mb-1GbE)",
|
||||
"q10": "120",
|
||||
"q11": "Average 150 feet, longest run approximately 280 feet to corner offices",
|
||||
"q12": ["Plenum-rated cable required"],
|
||||
"q13": ["Router/Firewall", "Unmanaged switches", "Wireless access points"],
|
||||
"q14": "Managed (Layer 2)",
|
||||
"q15": "6",
|
||||
"q16": "1800W estimated for all network equipment including PoE devices",
|
||||
"q17": ["IP Phones", "Wireless APs", "IP Cameras"],
|
||||
"q18": "Limited backup (1-2 hours)",
|
||||
"q19": "45",
|
||||
"q20": "85",
|
||||
"q21": ["Email/Web Browsing", "Video conferencing", "File sharing", "Cloud applications", "Remote desktop"],
|
||||
"q22": ["SOC 2"],
|
||||
"q23": ["Corporate network", "Guest network", "IoT network", "Management network"],
|
||||
"q24": ["Firewall", "VPN access", "Network access control (NAC)"],
|
||||
"q25": ["Fiber", "5G/Cellular backup"],
|
||||
"q26": "500Mbps download / 250Mbps upload",
|
||||
"q27": "No - Single site only",
|
||||
"q28": "Yes - Partial upgrade",
|
||||
"q29": "Current WiFi has dead zones in conference rooms. Existing switches are unmanaged and showing signs of age. Need better coverage and management capabilities.",
|
||||
"q30": ["Slow internet", "WiFi dead zones", "Network congestion"],
|
||||
"q31": "$25,000 - $50,000",
|
||||
"q32": "Within 3 months",
|
||||
"q33": "Internal IT team",
|
||||
"q34": "",
|
||||
"q35": "Need outdoor WiFi coverage for patio area. Plan to add IP cameras in future. Prefer Cisco or Ubiquiti equipment. Require weekend installation to minimize downtime."
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@app.route('/api/health', methods=['GET'])
|
||||
def health_check():
|
||||
"""Health check endpoint"""
|
||||
try:
|
||||
resp = requests.get(f"{OLLAMA_HOST}/api/tags", timeout=5)
|
||||
ollama_online = resp.status_code == 200
|
||||
except:
|
||||
ollama_online = False
|
||||
|
||||
return jsonify({
|
||||
"status": "ok",
|
||||
"service": "site-survey-ai",
|
||||
"timestamp": datetime.utcnow().isoformat(),
|
||||
"ollama_connected": ollama_online
|
||||
})
|
||||
|
||||
|
||||
@app.route('/api/models', methods=['GET'])
|
||||
def get_models():
|
||||
"""Get available Ollama models"""
|
||||
try:
|
||||
resp = requests.get(f"{OLLAMA_HOST}/api/tags", timeout=5)
|
||||
if resp.status_code == 200:
|
||||
models_data = resp.json()
|
||||
models = [m['name'] for m in models_data.get('models', [])]
|
||||
# Add cloud models if available
|
||||
cloud_models = ['kimi-k2.6:cloud', 'kimi-k2.5:cloud', 'nemotron-3-super:cloud']
|
||||
models.extend(cloud_models)
|
||||
return jsonify({"models": models, "default": "kimi-k2.6:cloud"})
|
||||
except Exception as e:
|
||||
print(f"Error fetching models: {e}")
|
||||
|
||||
return jsonify({
|
||||
"models": ["kimi-k2.6:cloud", "kimi-k2.5:cloud", "llava:latest"],
|
||||
"default": "kimi-k2.6:cloud"
|
||||
})
|
||||
|
||||
|
||||
@app.route('/api/ollama-status', methods=['GET'])
|
||||
def ollama_status():
|
||||
"""Check Ollama connection status"""
|
||||
try:
|
||||
resp = requests.get(f"{OLLAMA_HOST}/api/tags", timeout=5)
|
||||
return jsonify({"online": resp.status_code == 200})
|
||||
except:
|
||||
return jsonify({"online": False})
|
||||
|
||||
|
||||
@app.route('/api/surveys/template', methods=['GET'])
|
||||
def get_survey_template():
|
||||
"""Get the survey template with questions"""
|
||||
return jsonify({"template": DEFAULT_SURVEY_TEMPLATE})
|
||||
|
||||
|
||||
@app.route('/api/surveys', methods=['GET'])
|
||||
def list_surveys():
|
||||
"""List all surveys"""
|
||||
surveys = []
|
||||
for sid, survey in surveys_db.items():
|
||||
surveys.append({
|
||||
"id": sid,
|
||||
"name": survey['name'],
|
||||
"client_name": survey.get('client_name', ''),
|
||||
"site_name": survey.get('site_name', ''),
|
||||
"description": survey.get('description', ''),
|
||||
"created_at": survey['created_at'],
|
||||
"responses_count": len(survey.get('responses', []))
|
||||
})
|
||||
return jsonify({"surveys": surveys})
|
||||
|
||||
|
||||
@app.route('/api/surveys', methods=['POST'])
|
||||
def create_survey():
|
||||
"""Create a new survey"""
|
||||
data = request.get_json()
|
||||
|
||||
survey_id = str(uuid.uuid4())
|
||||
survey = {
|
||||
'id': survey_id,
|
||||
'name': data.get('name', 'New Survey'),
|
||||
'description': data.get('description', ''),
|
||||
'client_name': data.get('client_name', ''),
|
||||
'site_name': data.get('site_name', ''),
|
||||
'questions': DEFAULT_SURVEY_TEMPLATE['questions'],
|
||||
'created_at': datetime.utcnow().isoformat(),
|
||||
'status': 'active',
|
||||
'responses': []
|
||||
}
|
||||
|
||||
# If responses are provided, add them
|
||||
responses_data = data.get('responses')
|
||||
if responses_data and isinstance(responses_data, dict) and len(responses_data) > 0:
|
||||
response_entry = {
|
||||
'id': str(uuid.uuid4()),
|
||||
'survey_id': survey_id,
|
||||
'responses': responses_data,
|
||||
'photos': [],
|
||||
'analyzed': False,
|
||||
'analysis': None,
|
||||
'submitter_name': 'Survey Creator',
|
||||
'created_at': datetime.utcnow().isoformat()
|
||||
}
|
||||
survey['responses'].append(response_entry)
|
||||
|
||||
surveys_db[survey_id] = survey
|
||||
return jsonify({"survey": survey}), 201
|
||||
|
||||
|
||||
@app.route('/api/surveys/<survey_id>', methods=['GET'])
|
||||
def get_survey(survey_id):
|
||||
"""Get a specific survey"""
|
||||
if survey_id not in surveys_db:
|
||||
return jsonify({"error": "Survey not found"}), 404
|
||||
return jsonify({"survey": surveys_db[survey_id]})
|
||||
|
||||
|
||||
@app.route('/api/surveys/<survey_id>', methods=['DELETE'])
|
||||
def delete_survey(survey_id):
|
||||
"""Delete a survey"""
|
||||
if survey_id not in surveys_db:
|
||||
return jsonify({"error": "Survey not found"}), 404
|
||||
|
||||
del surveys_db[survey_id]
|
||||
return jsonify({"success": True, "message": "Survey deleted"})
|
||||
|
||||
|
||||
@app.route('/api/surveys', methods=['DELETE'])
|
||||
def delete_all_surveys():
|
||||
"""Delete all surveys"""
|
||||
count = len(surveys_db)
|
||||
surveys_db.clear()
|
||||
return jsonify({"success": True, "message": f"Deleted {count} surveys"})
|
||||
|
||||
|
||||
@app.route('/api/surveys/<survey_id>/responses', methods=['POST'])
|
||||
def submit_response(survey_id):
|
||||
"""Submit survey responses"""
|
||||
if survey_id not in surveys_db:
|
||||
return jsonify({"error": "Survey not found"}), 404
|
||||
|
||||
data = request.get_json()
|
||||
response_id = str(uuid.uuid4())
|
||||
|
||||
response_entry = {
|
||||
'id': response_id,
|
||||
'survey_id': survey_id,
|
||||
'responses': data.get('responses', {}),
|
||||
'photos': data.get('photos', []),
|
||||
'analyzed': False,
|
||||
'analysis': None,
|
||||
'submitter_name': data.get('submitter_name', 'Anonymous'),
|
||||
'created_at': datetime.utcnow().isoformat()
|
||||
}
|
||||
|
||||
surveys_db[survey_id]['responses'].append(response_entry)
|
||||
return jsonify({"response": response_entry}), 201
|
||||
|
||||
|
||||
@app.route('/api/surveys/<survey_id>/photos', methods=['POST'])
|
||||
def upload_photo(survey_id):
|
||||
"""Upload a photo for a survey"""
|
||||
if survey_id not in surveys_db:
|
||||
return jsonify({"error": "Survey not found"}), 404
|
||||
|
||||
if 'photo' not in request.files:
|
||||
return jsonify({"error": "No photo provided"}), 400
|
||||
|
||||
photo = request.files['photo']
|
||||
if photo.filename == '':
|
||||
return jsonify({"error": "No photo selected"}), 400
|
||||
|
||||
# Save photo
|
||||
filename = secure_filename(f"{survey_id}_{uuid.uuid4()}_{photo.filename}")
|
||||
photo_path = os.path.join(PHOTOS_FOLDER, filename)
|
||||
photo.save(photo_path)
|
||||
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"filename": filename,
|
||||
"url": f"/api/photos/{survey_id}/{filename}"
|
||||
})
|
||||
|
||||
|
||||
@app.route('/api/photos/<survey_id>/<filename>', methods=['GET'])
|
||||
def get_photo(survey_id, filename):
|
||||
"""Get a photo"""
|
||||
photo_path = os.path.join(PHOTOS_FOLDER, filename)
|
||||
if os.path.exists(photo_path):
|
||||
return send_file(photo_path)
|
||||
return jsonify({"error": "Photo not found"}), 404
|
||||
|
||||
|
||||
@app.route('/api/surveys/<survey_id>/analyze', methods=['POST'])
|
||||
def analyze_survey(survey_id):
|
||||
"""Analyze survey with AI"""
|
||||
if survey_id not in surveys_db:
|
||||
return jsonify({"error": "Survey not found"}), 404
|
||||
|
||||
survey = surveys_db[survey_id]
|
||||
data = request.get_json() or {}
|
||||
selected_model = data.get('model', 'llava:latest')
|
||||
|
||||
# Get the latest response
|
||||
if not survey['responses']:
|
||||
return jsonify({"error": "No responses to analyze"}), 400
|
||||
|
||||
last_response = survey['responses'][-1]
|
||||
responses = last_response['responses']
|
||||
|
||||
# Build analysis prompt
|
||||
prompt = f"""Analyze this IT site survey and provide recommendations:
|
||||
|
||||
Survey: {survey['name']}
|
||||
Client: {survey['client_name']}
|
||||
Site: {survey['site_name']}
|
||||
|
||||
Responses:
|
||||
"""
|
||||
|
||||
for q in survey['questions']:
|
||||
qid = q['id']
|
||||
if qid in responses:
|
||||
prompt += f"\n{q['question']}: {responses[qid]}"
|
||||
|
||||
prompt += "\n\nProvide a detailed analysis including:\n"
|
||||
prompt += "1. Network infrastructure recommendations\n"
|
||||
prompt += "2. Equipment recommendations with estimated costs\n"
|
||||
prompt += "3. Implementation timeline\n"
|
||||
prompt += "4. Potential challenges and solutions\n"
|
||||
|
||||
try:
|
||||
# Call Ollama for analysis
|
||||
ollama_data = {
|
||||
"model": selected_model,
|
||||
"messages": [{"role": "user", "content": prompt}],
|
||||
"stream": False
|
||||
}
|
||||
|
||||
resp = requests.post(OLLAMA_URL, json=ollama_data, timeout=300)
|
||||
if resp.status_code == 200:
|
||||
result = resp.json()
|
||||
analysis = result.get('message', {}).get('content', 'No analysis available')
|
||||
|
||||
# Save analysis
|
||||
last_response['analyzed'] = True
|
||||
last_response['analysis'] = analysis
|
||||
last_response['model_used'] = selected_model
|
||||
|
||||
return jsonify({
|
||||
"analysis": analysis,
|
||||
"model_used": selected_model
|
||||
})
|
||||
else:
|
||||
return jsonify({"error": f"Ollama error: {resp.status_code}"}), 500
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/api/surveys/<survey_id>/export', methods=['POST'])
|
||||
def export_survey(survey_id):
|
||||
"""Export survey results"""
|
||||
if survey_id not in surveys_db:
|
||||
return jsonify({"error": "Survey not found"}), 404
|
||||
|
||||
data = request.get_json() or {}
|
||||
export_format = data.get('format', 'json')
|
||||
|
||||
survey = surveys_db[survey_id]
|
||||
|
||||
if export_format == 'json':
|
||||
return jsonify({"survey": survey})
|
||||
elif export_format == 'text':
|
||||
# Generate text report
|
||||
text = f"IT Site Survey Report\n"
|
||||
text += f"Survey: {survey['name']}\n"
|
||||
text += f"Client: {survey['client_name']}\n"
|
||||
text += f"Site: {survey['site_name']}\n"
|
||||
text += f"Date: {survey['created_at']}\n\n"
|
||||
|
||||
for response in survey['responses']:
|
||||
text += f"Response by: {response['submitter_name']}\n"
|
||||
for q in survey['questions']:
|
||||
qid = q['id']
|
||||
if qid in response['responses']:
|
||||
text += f"{q['question']}: {response['responses'][qid]}\n"
|
||||
if response.get('analysis'):
|
||||
text += f"\nAI Analysis:\n{response['analysis']}\n"
|
||||
|
||||
return jsonify({"text": text})
|
||||
else:
|
||||
return jsonify({"error": "Unsupported format"}), 400
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host='0.0.0.0', port=3003, debug=False)
|
||||
Reference in New Issue
Block a user