Spaces:

NCEE-Build-Lab
/

ego_jimmy

Sleeping

App Files Files Community

MilanM commited on Sep 29, 2024

Commit

5b7fdba

verified ·

1 Parent(s): 2a856ec

Update app.py

Browse files

Files changed (1) hide show

app.py +83 -7

app.py CHANGED Viewed

@@ -5,8 +5,13 @@ import secretsload
 import anton_ego_jimmy
 import requests
 import time
 import re
 from ibm_watsonx_ai.foundation_models import ModelInference
 from ibm_watsonx_ai import Credentials, APIClient
 from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams
@@ -18,7 +23,8 @@ credentials = load_stsecrets()
 st.set_page_config(
     page_title="Jimmy",
     page_icon="🧐",
-    initial_sidebar_state="collapsed"
 )
 # Password protection
@@ -34,13 +40,13 @@ def check_password():
         st.markdown("\n\n")
         st.text_input("Enter the password", type="password", on_change=password_entered, key="password")
         st.divider()
-        st.info("Template of Jimmy developed by Milan Mrdenovic © IBM Norway 2024")
         return False
     elif not st.session_state["password_correct"]:
         st.markdown("\n\n")
         st.text_input("Enter the password", type="password", on_change=password_entered, key="password")
         st.divider()
-        st.info("Template of Jimmy developed by Milan Mrdenovic © IBM Norway 2024")
         st.error("😕 Password incorrect")
         return False
     else:
@@ -99,6 +105,58 @@ def generate_response(watsonx_llm, prompt_data, params):
     for chunk in generated_response:
         yield chunk
 def get_response(user_input):
     # Prepare the prompt
     prompt = prepare_prompt(user_input, st.session_state.chat_history)
@@ -141,13 +199,19 @@ def get_response(user_input):
 def main():
     initialize_session_state()
-    st.title("Jimmy 🧐")
     # User input
     if anton_ego_jimmy.DISPLAY_CHAT_HISTORY == 1:
-        for message in st.session_state.chat_history:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
     user_input = st.chat_input("You:", key="user_input")
     if user_input:
@@ -159,5 +223,17 @@ def main():
         # Get response
         get_response(user_input)
 if __name__ == "__main__":
     main()

 import anton_ego_jimmy
 import requests
 import time
+import regex
 import re
+from datetime import datetime
+import pdfkit
+from jinja2 import Template
 from ibm_watsonx_ai.foundation_models import ModelInference
 from ibm_watsonx_ai import Credentials, APIClient
 from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams
 st.set_page_config(
     page_title="Jimmy",
     page_icon="🧐",
+    initial_sidebar_state="collapsed",
+    layout="centered"
 )
 # Password protection
         st.markdown("\n\n")
         st.text_input("Enter the password", type="password", on_change=password_entered, key="password")
         st.divider()
+        st.info("Designed and developed by Milan Mrdenovic © IBM Norway 2024")
         return False
     elif not st.session_state["password_correct"]:
         st.markdown("\n\n")
         st.text_input("Enter the password", type="password", on_change=password_entered, key="password")
         st.divider()
+        st.info("Designed and developed by Milan Mrdenovic © IBM Norway 2024")
         st.error("😕 Password incorrect")
         return False
     else:
     for chunk in generated_response:
         yield chunk
+emoji_pattern = regex.compile(r'\p{Emoji}', flags=regex.UNICODE)
+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
     prompt = prepare_prompt(user_input, st.session_state.chat_history)
 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:
         # 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()