Spaces:

FrederickSundeep
/

AIHRTicketGenerator

Sleeping

App Files Files Community

FrederickSundeep commited on Apr 22, 2025

Commit

0c729b9

1 Parent(s): 3f21d3a

update file 007

Browse files

Files changed (3) hide show

app.py +55 -33
requirements.txt +1 -1
templates/index.html +56 -40

app.py CHANGED Viewed

@@ -1,55 +1,77 @@
-from flask import Flask, render_template, request, jsonify
-from transformers import pipeline, set_seed
 from langchain.prompts import PromptTemplate
-from fpdf import FPDF
 import os
 from datetime import datetime
 app = Flask(__name__)
-# Load lightweight LLM
-generator = pipeline('text-generation', model='distilgpt2')
-set_seed(42)
-# Prompt Template using LangChain
 template = PromptTemplate(
-    input_variables=["issue"],
-    template="Generate a formal HR ticket for the following issue:\n\n{issue}\n\nTicket:"
 )
-@app.route('/')
-def index():
-    return render_template('index.html')
-@app.route('/generate_ticket', methods=['POST'])
 def generate_ticket():
-    user_input = request.form['query']
-    prompt = template.format(issue=user_input)
-    result = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
-    ticket = result.replace(prompt, '').strip()
-    # Session logging
-    with open("session_log.txt", "a") as f:
-        f.write(f"\n[{datetime.now()}]\nIssue: {user_input}\nTicket: {ticket}\n")
-    return jsonify({'ticket': ticket})
-@app.route('/download_pdf', methods=['POST'])
-def download_pdf():
-    content = request.form['ticket']
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    for line in content.split('\n'):
-        pdf.multi_cell(0, 10, line)
-    filename = "hr_ticket.pdf"
-    pdf.output(filename)
-    return jsonify({'success': True, 'file': filename})
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)

+from flask import Flask, request, jsonify, send_file, render_template
+from transformers import pipeline
 from langchain.prompts import PromptTemplate
+from reportlab.pdfgen import canvas
 import os
+import uuid
 from datetime import datetime
 app = Flask(__name__)
+# Load lightweight model for CPU
+generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", device=-1)
+# Prompt template
 template = PromptTemplate(
+    input_variables=["issue", "name", "email"],
+    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.
+    """
 )
+# 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")
+    email = request.form.get("email")
+    issue = request.form.get("issue")
+    prompt = template.format(name=name, email=email, 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")
+    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")
+    return render_template("index.html", ticket=result, session_id=session_id)
+@app.route("/download/<session_id>")
+def download_pdf(session_id):
+    log_path = f"logs/{session_id}.txt"
+    pdf_path = f"pdfs/{session_id}.pdf"
+    if not os.path.exists(log_path):
+        return jsonify({"error": "Session not found"}), 404
+    with open(log_path, "r") as f:
+        content = f.read()
+    c = canvas.Canvas(pdf_path)
+    lines = content.splitlines()
+    y = 800
+    for line in lines:
+        c.drawString(50, y, line)
+        y -= 15
+    c.save()
+    return send_file(pdf_path, as_attachment=True)
 if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)

requirements.txt CHANGED Viewed

@@ -1,5 +1,5 @@
 flask
 transformers
 torch
-fpdf
 langchain

 flask
 transformers
 torch
 langchain
+reportlab

templates/index.html CHANGED Viewed

@@ -1,52 +1,68 @@
 <!DOCTYPE html>
 <html>
 <head>
-    <title>HR Ticket Generator</title>
     <style>
-        body { font-family: Arial; margin: 40px; }
-        input, button { padding: 10px; margin-top: 10px; width: 100%; }
-        textarea { width: 100%; height: 150px; margin-top: 20px; }
     </style>
 </head>
 <body>
-    <h2>HR Ticket Generator</h2>
-    <form id="ticketForm">
-        <input type="text" name="query" placeholder="Enter your HR issue..." required />
-        <button type="submit">Generate Ticket</button>
-    </form>
-    <textarea id="result" readonly></textarea>
-    <button id="downloadBtn" style="display:none;">Download as PDF</button>
-    <script>
-        const form = document.getElementById('ticketForm');
-        const resultBox = document.getElementById('result');
-        const downloadBtn = document.getElementById('downloadBtn');
-        form.onsubmit = async (e) => {
-            e.preventDefault();
-            const formData = new FormData(form);
-            const response = await fetch('/generate_ticket', {
-                method: 'POST',
-                body: formData
-            });
-            const data = await response.json();
-            resultBox.value = data.ticket;
-            downloadBtn.style.display = 'inline-block';
-        };
-        downloadBtn.onclick = async () => {
-            const ticket = resultBox.value;
-            const formData = new FormData();
-            formData.append('ticket', ticket);
-            const res = await fetch('/download_pdf', {
-                method: 'POST',
-                body: formData
-            });
-            const file = await res.json();
-            if (file.success) {
-                window.location.href = file.file;
-            }
-        };
-    </script>
 </body>
 </html>

 <!DOCTYPE html>
 <html>
 <head>
+    <title>AI HR Ticket Generator</title>
     <style>
+        body {
+            font-family: Arial, sans-serif;
+            padding: 40px;
+            background-color: #f0f2f5;
+        }
+        form {
+            background: white;
+            padding: 30px;
+            border-radius: 10px;
+            box-shadow: 0px 0px 8px rgba(0, 0, 0, 0.1);
+            max-width: 600px;
+            margin: auto;
+        }
+        input, textarea {
+            width: 100%;
+            padding: 12px;
+            margin: 10px 0 20px;
+            border: 1px solid #ccc;
+            border-radius: 8px;
+        }
+        button {
+            background-color: #007bff;
+            border: none;
+            padding: 12px 25px;
+            border-radius: 8px;
+            color: white;
+            font-size: 16px;
+        }
+        .ticket-box {
+            background: #e6ffe6;
+            padding: 20px;
+            margin-top: 30px;
+            border-radius: 10px;
+        }
     </style>
 </head>
 <body>
+<h2 align="center">🤖 AI HR Ticket Generator</h2>
+<form action="/generate" method="post">
+    <label for="name">Employee Name:</label>
+    <input type="text" name="name" required>
+    <label for="email">Email:</label>
+    <input type="email" name="email" required>
+    <label for="issue">Issue Description:</label>
+    <textarea name="issue" rows="5" required></textarea>
+    <button type="submit">Generate Ticket</button>
+</form>
+{% if ticket %}
+<div class="ticket-box">
+    <h3>Generated Ticket:</h3>
+    <pre>{{ ticket }}</pre>
+    <a href="/download/{{ session_id }}"><button>Download as PDF</button></a>
+</div>
+{% endif %}
 </body>
 </html>