Upload 4 files

Dockerfile

@@ -0,0 +1,36 @@
+# Use official Python slim image
+FROM python:3.11-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    git \
+    libglib2.0-0 \
+    libsm6 \
+    libxrender1 \
+    libxext6 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Use /data for all Hugging Face cache to avoid permission issues in Spaces
+ENV TRANSFORMERS_CACHE=/data/transformers
+ENV HF_HOME=/data/huggingface
+
+# Create writable cache directories
+RUN mkdir -p /data/transformers /data/huggingface && chmod -R 777 /data
+
+# Copy requirements and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install flask python-dotenv
+
+# Copy full app code
+COPY . .
+
+# Expose Flask port
+EXPOSE 7860
+
+# Run the Flask app
+CMD ["flask", "run", "--host=0.0.0.0", "--port=7860"]

app.py

@@ -0,0 +1,264 @@
+from unicodedata import category
+from flask import Flask, render_template, request, jsonify
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+import os
+import logging
+import requests
+import threading
+from queue import Queue
+from functools import lru_cache
+
+app = Flask(__name__)
+logging.basicConfig(level=logging.INFO)
+
+CACHE_DIR = "/data"
+model_name = "Qwen/Qwen2.5-0.5B-Instruct"
+
+# Load model and tokenizer
+logging.info("Loading tokenizer...")
+tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=CACHE_DIR, trust_remote_code=True)
+logging.info("Loading model...")
+model = AutoModelForCausalLM.from_pretrained(model_name, cache_dir=CACHE_DIR, trust_remote_code=True)
+logging.info("Creating pipeline...")
+generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=-1)
+logging.info("Model ready.")
+
+# Fallback description text
+def get_fallback_description(details):
+    category = details.get('parent_category', '').lower()
+    address = f"{details.get('address', '')}, {details.get('city', '')}, {details.get('state', '')}"
+    price = details.get('price', '')
+    rooms = int(details.get('total_rooms', 0)) if details.get('total_rooms') else 0
+    area = details.get('area', '')
+    
+    # Get nearby locations
+    nearby_info = ""
+    lat = details.get('latitude')
+    lon = details.get('longitude')
+    if lat and lon:
+        try:
+            lat_f, lon_f = float(lat), float(lon)
+            result_queue = Queue()
+            thread = threading.Thread(target=fetch_nearby_amenities_thread, args=(lat_f, lon_f, 2500, result_queue))
+            thread.start()
+            thread.join(timeout=8)
+            landmarks = result_queue.get() if not result_queue.empty() else []
+            
+            if landmarks:
+                amenities = [x for x in landmarks if any(term in x.lower() for term in ['school', 'hospital', 'restaurant', 'pharmacy', 'bank', 'atm', 'cafe', 'bus', 'fuel', 'supermarket', 'police', 'post'])]
+                landmarks = [x for x in landmarks if x not in amenities]
+                
+                if amenities:
+                    nearby_info += f" The property is conveniently located near {', '.join(amenities[:2])}."
+                if landmarks:
+                    nearby_info += f" Notable landmarks include {', '.join(landmarks[:2])}."
+        except Exception as e:
+            logging.error(f"Fallback amenity fetch error: {str(e)}")
+            pass
+    
+    if category in ['agricultural', 'commercial']:
+        return f'''A prime agricultural property located at {address}. This expansive land spans {area} sq. ft., making it ideal for farming or agricultural development. The property is priced at ₹{price} and features {rooms} storage/utility rooms. The land is well-irrigated and offers excellent soil quality, perfect for various agricultural activities.{nearby_info} Whether you're looking to start a farm or expand your agricultural operations, this property provides the perfect foundation for your agricultural ventures.'''
+    else:  # Commercial
+        return f'''A premium commercial property situated in the heart of {address}. This modern commercial space spans {area} sq. ft., offering a versatile layout with {rooms} rooms for optimal business operations. Priced at ₹{price}, this property features excellent visibility and accessibility.{nearby_info} The space is designed to accommodate various business needs, from retail to office use, with modern amenities and professional atmosphere. Whether you're establishing a new business or expanding existing operations, this location provides the perfect blend of convenience and functionality.'''
+
+@lru_cache(maxsize=100)
+def fetch_nearby_amenities_thread(lat, lon, radius, result_queue):
+    try:
+        overpass_url = "https://overpass-api.de/api/interpreter"
+        query = f"""
+        [out:json][timeout:10];
+        (
+          node["amenity"~"school|hospital|restaurant|temple|mandir|hotel|pharmacy|bank|atm|cafe|bus_station|fuel|supermarket|police|post_office"](around:{radius},{lat},{lon});
+          node["tourism"~"hotel|museum|gallery|viewpoint|attraction"](around:{radius},{lat},{lon});
+          node["historic"~"monument|memorial|archaeological_site|castle|ruins"](around:{radius},{lat},{lon});
+          node["leisure"~"park|garden|sports_centre|stadium"](around:{radius},{lat},{lon});
+        );
+        out body;
+        """
+        response = requests.post(overpass_url, data={'data': query}, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+        landmarks = []
+        seen_names = set()
+        
+        for element in data.get('elements', []):
+            tags = element.get('tags', {})
+            name = tags.get('name')
+            if not name or name in seen_names:
+                continue
+            seen_names.add(name)
+            
+            place_type = None
+            if tags.get('tourism'):
+                place_type = tags['tourism'].replace('_', ' ').title()
+            elif tags.get('historic'):
+                place_type = tags['historic'].replace('_', ' ').title()
+            elif tags.get('leisure'):
+                place_type = tags['leisure'].replace('_', ' ').title()
+            elif tags.get('amenity'):
+                place_type = tags['amenity'].replace('_', ' ').title()
+                
+            if place_type:
+                landmarks.append((f"{place_type} '{name}'", len(name)))
+        
+        # Sort by name length and take top results
+        landmarks.sort(key=lambda x: x[1], reverse=True)
+        result_queue.put([x[0] for x in landmarks])
+    except Exception as e:
+        logging.error(f"Overpass API error: {str(e)}")
+        result_queue.put([])
+
+def generate_description(details):
+    try:
+        category = details.get('parent_category', '').lower()
+        
+        # Basic info - only include beds/rooms for residential properties
+        extra_info = ""
+        if category not in ['agricultural', 'commercial']:
+            beds = int(details.get('beds', 0)) if details.get('beds') else 0
+            rooms = int(details.get('total_rooms', 0)) if details.get('total_rooms') else 0
+            baths = int(details.get('baths',0)) if details.get('baths') else 0
+            bed_info = f"{beds} bed{'s' if beds != 1 else ''}" if beds > 0 else ""
+            room_info = f"{rooms} room{'s' if rooms != 1 else ''}" if rooms > 0 else ""
+            bath_info = f"{baths} bath{'s' if baths !=1 else ''}" if baths > 0 else ""
+            if beds or rooms or baths:
+                parts = []
+                if beds: parts.append(bed_info)
+                if rooms: parts.append(room_info)
+                if baths: parts.append(bath_info)
+                extra_info = "This property includes " + " and ".join(parts) + "."
+
+        # Amenities thread with shorter timeout
+        amenities_str = "None found nearby"
+        landmarks_str = "None found nearby"
+        lat = details.get('latitude')
+        lon = details.get('longitude')
+        if lat and lon:
+            try:
+                lat_f, lon_f = float(lat), float(lon)
+                result_queue = Queue()
+                thread = threading.Thread(target=fetch_nearby_amenities_thread, args=(lat_f, lon_f, 2000, result_queue))
+                thread.start()
+                thread.join(timeout=8)
+                landmarks = result_queue.get() if not result_queue.empty() else []
+                
+                # Split into amenities and landmarks
+                amenities = [x for x in landmarks if any(term in x.lower() for term in ['school', 'hospital', 'restaurant', 'pharmacy', 'bank', 'atm', 'cafe', 'bus', 'fuel', 'supermarket', 'police', 'post'])]
+                landmarks = [x for x in landmarks if x not in amenities]
+                
+                amenities_str = ", ".join(amenities[:3]) if amenities else "None found nearby"
+                landmarks_str = ", ".join(landmarks[:3]) if landmarks else "None found nearby"
+            except Exception as e:
+                logging.error(f"Amenity fetch error: {e}")
+
+        # Prompt construction - exclude beds/rooms/baths for agricultural and commercial
+        base_fields = [
+            f"Property Name: {details.get('property_name', '')}",
+            f"Type: {details.get('property_type', '')}",
+            f"Category: {details.get('parent_category', '')}",
+            f"Address: {details.get('address', '')}, {details.get('city', '')}, {details.get('state', '')}, {details.get('zipcode', '')}, {details.get('country', '')}",
+            f"Coordinates: ({details.get('latitude', '')}, {details.get('longitude', '')})",
+            f"Year Built: {details.get('year', '')}",
+            f"Price: ₹{details.get('price', '')}",
+            f"Status: {details.get('property_status', '')}",
+            f"Code: {details.get('property_code', '')}",
+            f"Area: {details.get('area', '')} sq. ft.",
+            f"Features: {details.get('features', '')}",
+            f"Nearby Amenities: {amenities_str}",
+            f"Nearby Landmarks: {landmarks_str}"
+        ]
+
+        if category not in ['agricultural', 'commercial']:
+            base_fields.extend([
+                f"Rooms: {details.get('total_rooms', '')}",
+                f"Beds: {details.get('beds', '')}",
+                f"Baths: {details.get('baths', '')}"
+            ])
+
+        prompt = "Generate a property description between 150 and 200 words. Focus ONLY on property features, location, and amenities. Do NOT include any references to natural features like rivers, lakes, islands, or landscapes. Use ALL fields below. Do NOT skip any:\n- " + "\n- ".join(base_fields)
+
+        if extra_info:
+            prompt += f"\n\n{extra_info}"
+        prompt += "\nWrite a natural, complete property description focusing on urban and property characteristics only.\nDescription:"
+
+        # Run generation in thread with longer timeout
+        result_queue = Queue()
+        def run_generation(q):
+            try:
+                out = generator(prompt, max_new_tokens=200, temperature=0.5)[0]['generated_text']
+                q.put(out)
+            except Exception as e:
+                q.put(f"Error: {e}")
+
+        gen_thread = threading.Thread(target=run_generation, args=(result_queue,))
+        gen_thread.start()
+        gen_thread.join(timeout=45)  # Increased timeout to 45 seconds
+
+        # Return dynamic fallback if generation takes too long or fails
+        if result_queue.empty():
+            logging.warning("Generation timeout - using fallback description")
+            return get_fallback_description(details)
+
+        result = result_queue.get()
+        if result.startswith("Error:"):
+            logging.warning(f"Generation error - using fallback description: {result}")
+            return get_fallback_description(details)
+
+        return result.split('Description:')[-1].strip() if 'Description:' in result else result.strip()
+
+    except Exception as e:
+        logging.error(f"Generation error: {str(e)}")
+        return FALLBACK_DESCRIPTION
+
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    description = None
+    if request.method == 'POST':
+        fields = ['parent_category', 'property_type', 'property_name', 'property_code', 'property_status',
+                  'price', 'area', 'total_rooms', 'beds', 'baths', 'year', 'address', 'country',
+                  'state', 'city', 'zipcode', 'latitude', 'longitude', 'features']
+        details = {f: request.form.get(f, '') for f in fields}
+
+        if not details['property_name']:
+            description = "Error: Property name is required."
+        else:
+            description = generate_description(details)
+
+    return render_template('index.html', description=description)
+
+@app.route('/generate', methods=['POST'])
+def generate():
+    if not request.is_json:
+        return {"error": "Request must be JSON"}, 400
+    details = request.get_json()
+
+    if not details.get('property_name'):
+        return {"error": "Missing required field: property_name"}, 400
+
+    desc = generate_description(details)
+    category = details.get('parent_category', '').lower()
+    
+    # Conditionally include information based on property category
+    if category not in ['agricultural', 'commercial']:
+        info_lines = []
+        beds = details.get('beds', '')
+        rooms = details.get('total_rooms', '')
+        baths = details.get('baths', '')
+        if beds and beds != '0':
+            info_lines.append(f"Beds: {beds}")
+        if rooms and rooms != '0':
+            info_lines.append(f"Rooms: {rooms}")
+        if baths and baths != '0':
+            info_lines.append(f"Baths: {baths}")
+        info_lines.append(f"Year: {details.get('year', '')}")
+        info_lines.append(f"Price: ₹{details.get('price', '')}")
+        extra_info = "\n" + "\n".join(info_lines)
+    else:
+        extra_info = f"\nYear: {details.get('year', '')}\nPrice: ₹{details.get('price', '')}"
+    
+    return {"description": desc + extra_info}
+
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 7860))
+    logging.info(f"Starting app on port {port}")
+    app.run(host='0.0.0.0', port=port, threaded=True)

requirements.txt

@@ -0,0 +1,13 @@
+streamlit
+langchain
+huggingface_hub
+torch
+transformers 
+langchain_community
+flask
+python-dotenv
+torch
+transformers
+accelerate
+sentencepiece
+requests

templates/index.html

@@ -0,0 +1,259 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI Property Description Generator</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+            background-color: #f5f5f5;
+        }
+        .container {
+            background-color: white;
+            padding: 30px;
+            border-radius: 10px;
+            box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+        }
+        h1 {
+            color: #2c3e50;
+            text-align: center;
+            margin-bottom: 30px;
+        }
+        .form-group {
+            margin-bottom: 20px;
+        }
+        label {
+            display: block;
+            margin-bottom: 5px;
+            color: #34495e;
+        }
+        select, input, textarea {
+            width: 100%;
+            padding: 8px;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            box-sizing: border-box;
+        }
+        button {
+            background-color: #3498db;
+            color: white;
+            padding: 10px 20px;
+            border: none;
+            border-radius: 4px;
+            cursor: pointer;
+            width: 100%;
+        }
+        button:hover {
+            background-color: #2980b9;
+        }
+        .description {
+            margin-top: 20px;
+            padding: 15px;
+            background-color: #f8f9fa;
+            border-radius: 4px;
+            border-left: 4px solid #3498db;
+        }
+        .form-row {
+            display: flex;
+            gap: 20px;
+            margin-bottom: 20px;
+        }
+        .form-group {
+            flex: 1;
+        }
+        .price-input {
+            display: flex;
+            align-items: center;
+        }
+        .price-input span {
+            padding: 8px;
+            background: #f8f9fa;
+            border: 1px solid #ddd;
+            border-right: none;
+            border-radius: 4px 0 0 4px;
+        }
+        .price-input input {
+            border-left: none;
+            border-radius: 0 4px 4px 0;
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>🏡 AI Property Description Generator</h1>
+        <form method="POST">
+            <div class="form-group">
+                <label for="parent_category">Parent Category:</label>
+                <select name="parent_category" id="parent_category" required>
+                    <option value="">Select Parent Category</option>
+                    <option value="Residential">Residential</option>
+                    <option value="Commercial">Commercial</option>
+                    <option value="Agricultural">Agricultural</option>
+                    <option value="PG/Co-living">PG/Co-living</option>
+                </select>
+            </div>
+
+            <div class="form-group">
+                <label for="property_type">Property Type:</label>
+                <select name="property_type" id="property_type" required>
+                    <option value="">Select Property Type</option>
+                    <option value="Townhouse">Townhouse</option>
+                    <option value="Flat">Flat</option>
+                    <option value="Villa">Villa</option>
+                    <option value="Office Space">Office Space</option>
+                    <option value="Shop">Shop</option>
+                    <option value="Farmland">Farmland</option>
+                    <option value="Commercial Land">Commercial Land</option>
+                    <option value="Plot/Land">Plot/Land</option>
+                    <option value="Boys Hostel">Boys Hostel</option>
+                    <option value="Girls Hostel">Girls Hostel</option>
+                    <option value="Working Womens Hostel">Working Women's Hostel</option>
+                    <option value="Boys PG">Boys PG</option>
+                    <option value="Girls PG">Girls PG</option>
+                    <option value="Showrooms">Showrooms</option>
+                    <option value="Co-working">Co-working</option>
+                </select>
+            </div>
+
+            <div class="form-row">
+                <div class="form-group">
+                    <label for="property_name">Property Name:</label>
+                    <input type="text" name="property_name" id="property_name" required>
+                </div>
+                <div class="form-group">
+                    <label for="property_code">Property Code:</label>
+                    <input type="text" name="property_code" id="property_code" value="Pr-00483" readonly>
+                </div>
+            </div>
+
+            <div class="form-group">
+                <label for="property_status">Property Status:</label>
+                <select name="property_status" id="property_status" required>
+                    <option value="">Select Status</option>
+                    <option value="Ready to move">Ready to move</option>
+                    <option value="Available">Available</option>
+                    <option value="Immediately">Immediately</option>
+                    <option value="Out of Stock">Out of Stock</option>
+                </select>
+            </div>
+
+            <div class="form-row">
+                <div class="form-group">
+                    <label for="price">Property Price:</label>
+                    <div class="price-input">
+                        <span>₹</span>
+                        <input type="number" name="price" id="price" placeholder="Enter property price" required>
+                    </div>
+                </div>
+                <div class="form-group">
+                    <label for="area">Area (sq ft):</label>
+                    <input type="number" name="area" id="area" placeholder="Enter area" required>
+                </div>
+            </div>
+
+            <div class="form-row">
+                <div class="form-group">
+                    <label for="total_rooms">Total Rooms:</label>
+                    <input type="number" name="total_rooms" id="total_rooms" required>
+                </div>
+                <div class="form-group">
+                    <label for="beds">Beds:</label>
+                    <input type="number" name="beds" id="beds" required>
+                </div>
+                <div class="form-group">
+                    <label for="baths">Baths:</label>
+                    <input type="number" name="baths" id="baths" required>
+                </div>
+            </div>
+
+            <div class="form-group">
+                <label for="year">Year Established:</label>
+                <input type="number" name="year" id="year" min="1900" max="2025" placeholder="Enter year (1900-2025)" required>
+            </div>
+
+            <div class="form-group">
+                <label for="address">Address:</label>
+                <textarea name="address" id="address" rows="3" placeholder="Address of your property" required></textarea>
+            </div>
+
+            <div class="form-row">
+                <div class="form-group">
+                    <label for="country">Country:</label>
+                    <input type="text" name="country" id="country" required>
+                </div>
+                <div class="form-group">
+                    <label for="state">State:</label>
+                    <input type="text" name="state" id="state" required>
+                </div>
+                <div class="form-group">
+                    <label for="city">City:</label>
+                    <input type="text" name="city" id="city" required>
+                </div>
+            </div>
+
+            <div class="form-row">
+                <div class="form-group">
+                    <label for="zipcode">Zip Code:</label>
+                    <input type="text" name="zipcode" id="zipcode" value="39702" required>
+                </div>
+                <div class="form-group">
+                    <label for="latitude">Latitude:</label>
+                    <input type="number" name="latitude" id="latitude" value="0" step="any">
+                </div>
+                <div class="form-group">
+                    <label for="longitude">Longitude:</label>
+                    <input type="number" name="longitude" id="longitude" value="0" step="any">
+                </div>
+            </div>
+
+            <div class="form-group">
+                <label for="features">Key Features:</label>
+                <textarea name="features" id="features" rows="4" placeholder="e.g. modern kitchen, parking, garden view" required></textarea>
+            </div>
+
+            <button type="button" id="generate-btn">Generate Description</button>
+        </form>
+        <div id="loading" style="display:none; color:#2980b9; margin-top:10px;">Generating description...</div>
+        <div id="error" style="display:none; color:red; margin-top:10px;"></div>
+        <div id="description-container" style="display:none;" class="description">
+            <h3>Generated Description:</h3>
+            <p id="description-text"></p>
+        </div>
+        <script>
+        document.getElementById('generate-btn').addEventListener('click', async function() {
+            const form = document.querySelector('form');
+            const formData = new FormData(form);
+            const data = {};
+            formData.forEach((value, key) => { data[key] = value; });
+            document.getElementById('loading').style.display = 'block';
+            document.getElementById('error').style.display = 'none';
+            document.getElementById('description-container').style.display = 'none';
+            try {
+                const response = await fetch('/generate', {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify(data)
+                });
+                const result = await response.json();
+                document.getElementById('loading').style.display = 'none';
+                if (response.ok && result.description) {
+                    document.getElementById('description-text').textContent = result.description;
+                    document.getElementById('description-container').style.display = 'block';
+                } else {
+                    document.getElementById('error').textContent = result.error || 'Failed to generate description.';
+                    document.getElementById('error').style.display = 'block';
+                }
+            } catch (e) {
+                document.getElementById('loading').style.display = 'none';
+                document.getElementById('error').textContent = 'An error occurred.';
+                document.getElementById('error').style.display = 'block';
+            }
+        });
+        </script>
+    </div>
+</body>
+</html>