Hugging Face's logo

Spaces:
hello674
/
sivfastapi
Sleeping

App Files Community
sivfastapi
File size: 5,270 Bytes
e7f02ca
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
# utils/pdf_generator.py pdf generator module
from fpdf import FPDF
from datetime import datetime
import os
import json
import re


class PDFGenerator:
    def __init__(self):
        # No initialization needed for this class
        pass
    def clean_text(self, text: str) -> str:
        replacements = {
            "→": "->",
            "–": "-",
            "—": "-",
            "…": "...",
        }
        for old, new in replacements.items():
            text = text.replace(old, new)
        return re.sub(r'[^\x00-\x7F]+', '', str(text))
    # At the top of utils/pdf_generator.py
    def safe_load(self, data):
        """Convert JSON string to dict if needed."""
        if isinstance(data, str):
            try:
                return json.loads(data)
            except json.JSONDecodeError:
                return {}
        return data

    def generate_pdf(self,idea: str, structured: dict, scores: dict, suggestions: dict, feedback: dict,chats: list, user_id: str) -> str:
        # Convert inputs
        structured = self.safe_load(structured)
        scores = self.safe_load(scores)
        suggestions = self.safe_load(suggestions)
        feedback = self.safe_load(feedback)

        pdf = FPDF()
        pdf.set_auto_page_break(auto=True, margin=15)
        pdf.add_page()
        pdf.set_font("Arial", 'B', 20)
        pdf.cell(0, 100, "Startup Idea Evaluation Report", ln=True, align='C')
        pdf.set_font("Arial", '', 14)
        pdf.cell(0, 10, f"Idea Name: {idea}", ln=True, align='C')
        pdf.ln(10)
        pdf.set_font("Arial", 'I', 12)
        pdf.cell(0, 10, f"Generated on {datetime.now().strftime('%Y-%m-%d')}", ln=True, align='C')
        pdf.add_page()  # then continue with details


        # Structured Format
        pdf.set_font("Arial", 'B', 12)
        pdf.cell(0, 10, "Structured Format:", ln=True)
        pdf.set_font("Arial", '', 11)
        for key, value in structured.items():
            pdf.multi_cell(0, 8, f"- {key.capitalize()}: {self.clean_text(value)}")
        pdf.ln(5)

        # Scores
        pdf.set_font("Arial", 'B', 12)
        pdf.cell(0, 10, "Scores:", ln=True)
        pdf.set_font("Arial", '', 11)
        for key, value in scores.items():
            pdf.multi_cell(0, 8, f"- {key}: {value}")
        pdf.ln(5)

        # Suggestions
        pdf.set_font("Arial", 'B', 12)
        pdf.cell(0, 10, "Suggestions:", ln=True)
        pdf.set_font("Arial", '', 11)
        for i, suggestion in enumerate(suggestions.get("improvements", []), start=1):
            pdf.multi_cell(0, 8, f"{i}. {self.clean_text(suggestion)}")
        pdf.ln(2)
        pdf.set_font("Arial", 'I', 10)
        for i, rationale in enumerate(suggestions.get("rationale", []), start=1):
            pdf.multi_cell(0, 7, f"   Rationale {i}: {self.clean_text(rationale)}")
        pdf.ln(5)

        # Feedback
        pdf.set_font("Arial", 'B', 12)
        pdf.cell(0, 10, "Feedback:", ln=True)
        pdf.set_font("Arial", '', 11)
        for i, fb in enumerate(feedback.get("feedbacks", []), start=1):
            pdf.multi_cell(0, 8, f"{i}. {self.clean_text(fb)}")
            pdf.ln(2)
        
        pdf.add_page()
        pdf.set_font("Arial", 'B', 16)
        pdf.cell(0, 10, "Expert Chats", ln=True, align='C')
        pdf.ln(5)

        if chats and len(chats) > 0:
            for expert in chats:
                pdf.set_font("Arial", 'B', 13)
                pdf.cell(0, 8, f"Expert: {expert.get('name', '')} ({expert.get('expertise', '')})", ln=True)
                pdf.set_font("Arial", 'I', 11)
                pdf.multi_cell(0, 6, f"Bio: {self.clean_text(expert.get('bio', ''))}")
                pdf.multi_cell(0, 6, f"Email: {expert.get('email', '')}")
                pdf.ln(3)

                chat = expert.get("chat", {})
                chathistory = chat.get("chathistory", [])

                if chathistory:
                    pdf.set_font("Arial", '', 11)
                    for msg in chathistory:
                        timestamp = datetime.fromisoformat(msg["timestamp"].replace("Z", "+00:00")).strftime("%Y-%m-%d %H:%M")
                        sender = msg.get("sender", "unknown").capitalize()
                        message = self.clean_text(msg.get("message", ""))
                        pdf.multi_cell(0, 8, f"[{timestamp}] {sender}: {message}")
                    pdf.ln(6)
                else:
                    pdf.set_font("Arial", 'I', 11)
                    pdf.cell(0, 8, "No messages found.", ln=True)
                    pdf.ln(5)
        else:
            pdf.set_font("Arial", 'I', 12)
            pdf.cell(0, 10, "No expert chats available.", ln=True, align='C')

        # Footer
        pdf.set_y(-30)
        pdf.set_font("Arial", 'I', 8)
        pdf.cell(0, 10, f"Generated by SIV | Version: {4.0} | Date: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", align='C')

        # Save file
        output_dir = "/mnt/data"
        os.makedirs(output_dir, exist_ok=True)
        output_path = os.path.join(output_dir, f"{user_id}_startup_report.pdf")
        pdf.output(output_path)

        return output_path