update file 013

app.py

@@ -2,55 +2,108 @@ from flask import Flask, request, jsonify, send_file, render_template
 from transformers import pipeline
 from langchain.prompts import PromptTemplate
 from reportlab.pdfgen import canvas
+from datetime import datetime
 import os
 import uuid
-from datetime import datetime
+import sqlite3
 
 app = Flask(__name__)
 
-# Load lightweight model for CPU
-# generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=-1)
-generator = pipeline("text-generation", model="gpt2", device=-1)
+# Load model (CPU)
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=-1)
 
+# Directories
+os.makedirs("logs", exist_ok=True)
+os.makedirs("pdfs", exist_ok=True)
+os.makedirs("data", exist_ok=True)
+
+# DB setup
+DB_PATH = "data/tickets.db"
+
+def init_db():
+    with sqlite3.connect(DB_PATH) as conn:
+        cursor = conn.cursor()
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS tickets (
+                id TEXT PRIMARY KEY,
+                first_name TEXT,
+                last_name TEXT,
+                email TEXT,
+                issue TEXT,
+                priority TEXT,
+                date TEXT,
+                generated TEXT,
+                created_at TEXT
+            )
+        ''')
+        conn.commit()
+
+init_db()
 
 # Prompt template
 template = PromptTemplate(
-    input_variables=["issue", "name", "email"],
+    input_variables=["first_name", "last_name", "email", "date", "priority", "issue"],
     template="""
-    You are an HR support assistant. Generate a professional HR helpdesk ticket based on the issue.
-    
-    Employee Name: {name}
-    Email: {email}
-    Reported Issue: {issue}
-
-    Provide a ticket title and description.
-    """
+You are an HR support assistant. Generate a professional HR helpdesk ticket.
+
+Employee Details:
+- First Name: {first_name}
+- Last Name: {last_name}
+- Email: {email}
+- Issue Date: {date}
+- Priority: {priority}
+- Reported Issue: {issue}
+
+Write a clear, concise ticket title and description suitable for HR review.
+"""
 )
 
-# Directories
-os.makedirs("logs", exist_ok=True)
-os.makedirs("pdfs", exist_ok=True)
-
 @app.route("/")
 def home():
     return render_template("index.html")
 
 @app.route("/generate", methods=["POST"])
 def generate_ticket():
-    name = request.form.get("name")
+    first_name = request.form.get("firstName")
+    last_name = request.form.get("lastName")
     email = request.form.get("email")
     issue = request.form.get("issue")
-    print(f"Received: Name={name}, Email={email}, Issue={issue}")
-    prompt = template.format(name=name, email=email, issue=issue)
+    priority = request.form.get("priority")
+    date = request.form.get("date")
+
+    prompt = template.format(
+        first_name=first_name,
+        last_name=last_name,
+        email=email,
+        date=date,
+        priority=priority,
+        issue=issue
+    )
+
     result = generator(prompt, max_length=200, do_sample=True)[0]["generated_text"]
 
     session_id = str(uuid.uuid4())
-    timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
+    # Save log file
     log_path = f"logs/{session_id}.txt"
     with open(log_path, "w") as f:
         f.write(f"Timestamp: {timestamp}\n")
-        f.write(f"Name: {name}\nEmail: {email}\nIssue: {issue}\n\nGenerated:\n{result}\n")
+        f.write(f"First Name: {first_name}\nLast Name: {last_name}\n")
+        f.write(f"Email: {email}\nDate: {date}\nPriority: {priority}\n")
+        f.write(f"Issue: {issue}\n\nGenerated:\n{result}\n")
+
+    # Save to SQLite DB
+    with sqlite3.connect(DB_PATH) as conn:
+        cursor = conn.cursor()
+        cursor.execute('''
+            INSERT INTO tickets (id, first_name, last_name, email, issue, priority, date, generated, created_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ''', (
+            session_id, first_name, last_name, email,
+            issue, priority, date, result, timestamp
+        ))
+        conn.commit()
 
     return render_template("index.html", ticket=result, session_id=session_id)
 
@@ -75,5 +128,33 @@ def download_pdf(session_id):
 
     return send_file(pdf_path, as_attachment=True)
 
+@app.route("/tickets")
+def view_tickets():
+    with sqlite3.connect(DB_PATH) as conn:
+        cursor = conn.cursor()
+        cursor.execute('''
+            SELECT id, first_name, last_name, email, priority, date, generated
+            FROM tickets
+            ORDER BY created_at DESC
+        ''')
+        rows = cursor.fetchall()
+
+    # Convert to list of dicts for easy templating
+    tickets = [
+        {
+            "id": row[0],
+            "first_name": row[1],
+            "last_name": row[2],
+            "email": row[3],
+            "priority": row[4],
+            "date": row[5],
+            "summary": (row[6][:100] + '...') if len(row[6]) > 100 else row[6]
+        }
+        for row in rows
+    ]
+
+    return render_template("tickets.html", tickets=tickets)
+
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)

templates/index.html

@@ -64,5 +64,9 @@
 </div>
 {% endif %}
 
+<p style="text-align: center;">
+    <a href="/tickets">📄 View Submitted Tickets</a>
+</p>
+
 </body>
 </html>

templates/tickets.html

@@ -0,0 +1,64 @@
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Submitted Tickets</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            padding: 40px;
+            background-color: #f9f9f9;
+        }
+        table {
+            width: 100%;
+            border-collapse: collapse;
+            margin-top: 30px;
+            background: white;
+        }
+        th, td {
+            padding: 12px;
+            text-align: left;
+            border-bottom: 1px solid #ddd;
+        }
+        th {
+            background: #007bff;
+            color: white;
+        }
+        a.button {
+            text-decoration: none;
+            padding: 8px 15px;
+            background: #28a745;
+            color: white;
+            border-radius: 6px;
+        }
+    </style>
+</head>
+<body>
+    <h2>📋 Submitted Tickets</h2>
+    <table>
+        <thead>
+            <tr>
+                <th>First</th>
+                <th>Last</th>
+                <th>Email</th>
+                <th>Priority</th>
+                <th>Issue Date</th>
+                <th>Summary</th>
+                <th>PDF</th>
+            </tr>
+        </thead>
+        <tbody>
+        {% for ticket in tickets %}
+            <tr>
+                <td>{{ ticket.first_name }}</td>
+                <td>{{ ticket.last_name }}</td>
+                <td>{{ ticket.email }}</td>
+                <td>{{ ticket.priority }}</td>
+                <td>{{ ticket.date }}</td>
+                <td>{{ ticket.summary }}</td>
+                <td><a class="button" href="/download/{{ ticket.id }}">Download</a></td>
+            </tr>
+        {% endfor %}
+        </tbody>
+    </table>
+</body>
+</html>