Spaces:

NCEE-Build-Lab
/

ego_jimmy

Running

App Files Files Community

MilanM commited on Sep 29, 2024

Commit

722ee9c

verified ·

1 Parent(s): a2ac7d3

Update app.py

Browse files

Files changed (1) hide show

app.py +45 -66

app.py CHANGED Viewed

@@ -9,7 +9,6 @@ import regex
 import re
 from datetime import datetime
-import pdfkit
 from jinja2 import Template
 from ibm_watsonx_ai.foundation_models import ModelInference
@@ -18,6 +17,13 @@ from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams
 from ibm_watsonx_ai.metanames import GenTextReturnOptMetaNames as RetParams
 from secretsload import load_stsecrets # script to load credentials from HuggingFace secrets section
 credentials = load_stsecrets()
 st.set_page_config(
@@ -55,7 +61,6 @@ def check_password():
 if not check_password():
     st.stop()
 # Initialize session state
 if 'current_page' not in st.session_state:
     st.session_state.current_page = 0
@@ -111,51 +116,30 @@ def remove_emojis(text):
     return emoji_pattern.sub(r'', text)
 def create_pdf_from_chat(chat_history):
-    html_template = """
-    <html>
-    <head>
-        <meta charset="UTF-8">
-        <style>
-            body { font-family: Arial, sans-serif; }
-            .message { margin: 10px 0; padding: 10px; border-radius: 10px; }
-            .user { background-color: #b0b4f5; text-align: right; }
-            .jimmy { background-color: #d9b0f5; }
-        </style>
-    </head>
-    <body>
-        <h1>Chat History - Generated on {{ timestamp }}</h1>
-        {% for message in chat_history %}
-        <div class="message {{ message.role }}">
-            <strong>{{ message.role|capitalize }}:</strong> {{ message.content }}
-        </div>
-        {% endfor %}
-    </body>
-    </html>
-    """
-    template = Template(html_template)
-    # Remove emojis from chat history
-    emoji_free_chat_history = [
-        {"role": message["role"], "content": remove_emojis(message["content"])}
-        for message in chat_history
-    ]
-    html_content = template.render(
-        chat_history=emoji_free_chat_history,
-        timestamp=datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    )
-    pdf_options = {
-        'page-size': 'A4',
-        'margin-top': '0.75in',
-        'margin-right': '0.75in',
-        'margin-bottom': '0.75in',
-        'margin-left': '0.75in',
-        'encoding': 'UTF-8',
-    }
-    pdf = pdfkit.from_string(html_content, False, options=pdf_options)
-    return BytesIO(pdf)
 def get_response(user_input):
     # Prepare the prompt
@@ -180,7 +164,6 @@ def get_response(user_input):
         GenParams.DECODING_METHOD: anton_ego_jimmy.DECODING_METHOD,
         GenParams.MAX_NEW_TOKENS: anton_ego_jimmy.MAX_NEW_TOKENS,
         GenParams.MIN_NEW_TOKENS: anton_ego_jimmy.MIN_NEW_TOKENS,
-        # GenParams.LENGTH_PENALTY: anton_ego_jimmy.LENGTH_PENALTY,
         GenParams.REPETITION_PENALTY: anton_ego_jimmy.REPETITION_PENALTY,
         GenParams.STOP_SEQUENCES: anton_ego_jimmy.STOP_SEQUENCES
     }
@@ -201,17 +184,10 @@ def main():
     initialize_session_state()
     st.subheader("Jimmy 🧐")
-    # col1, col2 = st.columns(2)
-    # with col1:
-    # User input
-        # with st.container(height=500,border=True):
     if anton_ego_jimmy.DISPLAY_CHAT_HISTORY == 1:
-            for message in st.session_state.chat_history:
-                # with st.chat_message(message["role"]):
-                with st.chat_message(message["role"], avatar="🥷🏻" if message["role"] == "user" else "🧐"):
-                    st.markdown(message["content"])
     user_input = st.chat_input("You:", key="user_input")
     if user_input:
@@ -223,17 +199,20 @@ def main():
         # Get response
         get_response(user_input)
-    # if st.session_state.chat_history:
-    # with col2:
         now = datetime.now()
         date_str = now.strftime("%Y-%m-%d")
-        pdf_buffer = create_pdf_from_chat(st.session_state.chat_history)
-        st.download_button(
-            label="Download Chat History as PDF",
-            data=pdf_buffer,
-            file_name=f"chat_history_{date_str}.pdf",
-            mime="application/pdf"
-        )
 if __name__ == "__main__":
     main()

 import re
 from datetime import datetime
 from jinja2 import Template
 from ibm_watsonx_ai.foundation_models import ModelInference
 from ibm_watsonx_ai.metanames import GenTextReturnOptMetaNames as RetParams
 from secretsload import load_stsecrets # script to load credentials from HuggingFace secrets section
+# New imports for ReportLab
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib import colors
+from reportlab.lib.enums import TA_LEFT, TA_RIGHT
 credentials = load_stsecrets()
 st.set_page_config(
 if not check_password():
     st.stop()
 # Initialize session state
 if 'current_page' not in st.session_state:
     st.session_state.current_page = 0
     return emoji_pattern.sub(r'', text)
 def create_pdf_from_chat(chat_history):
+    buffer = BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=letter, topMargin=30, bottomMargin=30)
+    styles = getSampleStyleSheet()
+    flowables = []
+    title_style = ParagraphStyle('Title', parent=styles['Heading1'], fontSize=18, spaceAfter=20)
+    flowables.append(Paragraph(f"Chat History - Generated on {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", title_style))
+    user_style = ParagraphStyle('UserStyle', parent=styles['Normal'],
+                                backColor=colors.lightblue, borderPadding=10,
+                                alignment=TA_RIGHT)
+    jimmy_style = ParagraphStyle('JimmyStyle', parent=styles['Normal'],
+                                 backColor=colors.lavender, borderPadding=10)
+    for message in chat_history:
+        role = message["role"]
+        content = remove_emojis(message["content"])
+        style = user_style if role == "user" else jimmy_style
+        flowables.append(Paragraph(f"<b>{role.capitalize()}:</b> {content}", style))
+        flowables.append(Spacer(1, 12))
+    doc.build(flowables)
+    buffer.seek(0)
+    return buffer
 def get_response(user_input):
     # Prepare the prompt
         GenParams.DECODING_METHOD: anton_ego_jimmy.DECODING_METHOD,
         GenParams.MAX_NEW_TOKENS: anton_ego_jimmy.MAX_NEW_TOKENS,
         GenParams.MIN_NEW_TOKENS: anton_ego_jimmy.MIN_NEW_TOKENS,
         GenParams.REPETITION_PENALTY: anton_ego_jimmy.REPETITION_PENALTY,
         GenParams.STOP_SEQUENCES: anton_ego_jimmy.STOP_SEQUENCES
     }
     initialize_session_state()
     st.subheader("Jimmy 🧐")
     if anton_ego_jimmy.DISPLAY_CHAT_HISTORY == 1:
+        for message in st.session_state.chat_history:
+            with st.chat_message(message["role"], avatar="🥷🏻" if message["role"] == "user" else "🧐"):
+                st.markdown(message["content"])
     user_input = st.chat_input("You:", key="user_input")
     if user_input:
         # Get response
         get_response(user_input)
+    if st.session_state.chat_history:
         now = datetime.now()
         date_str = now.strftime("%Y-%m-%d")
+        try:
+            pdf_buffer = create_pdf_from_chat(st.session_state.chat_history)
+            st.download_button(
+                label="Download Chat History as PDF",
+                data=pdf_buffer,
+                file_name=f"chat_history_{date_str}.pdf",
+                mime="application/pdf"
+            )
+        except Exception as e:
+            st.error(f"An error occurred while generating the PDF: {str(e)}")
+            st.error("If this persists, please contact support.")
 if __name__ == "__main__":
     main()