Spaces:

pentarosarium
/

processor

Sleeping

App Files Files Community

pentarosarium commited on Oct 22, 2024

Commit

a6e9232

1 Parent(s): 4fa1abd

progress more (3.4)

Browse files

Files changed (2) hide show

app.py +34 -15
requirements.txt +2 -1

app.py CHANGED Viewed

@@ -7,17 +7,17 @@ import io
 from rapidfuzz import fuzz
 import os
 from openpyxl import load_workbook
-from langchain_community.chat_models import ChatOpenAI
 from langchain.prompts import PromptTemplate
 from langchain_core.runnables import RunnablePassthrough
 from transformers import pipeline
 from io import StringIO, BytesIO
 import sys
 import contextlib
-from fpdf import FPDF
 def display_sentiment_results(row, sentiment, impact=None, reasoning=None):
     if sentiment == "Negative":
@@ -46,7 +46,7 @@ def display_sentiment_results(row, sentiment, impact=None, reasoning=None):
     st.write("---")
 class StreamlitCapture:
     def __init__(self):
@@ -56,17 +56,36 @@ class StreamlitCapture:
         self.texts.append(str(text))
 def save_streamlit_output_to_pdf(texts):
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.add_font('DejaVu', '', 'DejaVuSansCondensed.ttf', uni=True)
-    pdf.set_font('DejaVu', '', 12)
-    for text in texts:
-        # Clean and encode the text
-        clean_text = text.encode('latin-1', errors='replace').decode('latin-1')
-        pdf.multi_cell(0, 10, clean_text)
-    pdf.output("result.pdf", 'F')
 # Initialize sentiment analyzers
@@ -139,7 +158,7 @@ def init_langchain_llm():
         llm = ChatOpenAI(
             base_url="https://api.groq.com/openai/v1",
             model="llama-3.1-70b-versatile",
-            api_key=groq_api_key,
             temperature=0.0
         )
         return llm
@@ -401,7 +420,7 @@ def main():
         unsafe_allow_html=True
     )
-    st.title("::: анализ мониторинга новостей СКАН-ИНТЕРФАКС (v.3.3):::")
     if 'processed_df' not in st.session_state:
         st.session_state.processed_df = None

 from rapidfuzz import fuzz
 import os
 from openpyxl import load_workbook
 from langchain.prompts import PromptTemplate
 from langchain_core.runnables import RunnablePassthrough
 from transformers import pipeline
 from io import StringIO, BytesIO
 import sys
 import contextlib
+from langchain_openai import ChatOpenAI  # Updated import
+import pdfkit
+from jinja2 import Template
 def display_sentiment_results(row, sentiment, impact=None, reasoning=None):
     if sentiment == "Negative":
     st.write("---")
 class StreamlitCapture:
     def __init__(self):
         self.texts.append(str(text))
 def save_streamlit_output_to_pdf(texts):
+    # Create HTML content
+    html_content = """
+    <html>
+    <head>
+        <meta charset="UTF-8">
+        <style>
+            body { font-family: Arial, sans-serif; }
+            .content { margin: 20px; }
+        </style>
+    </head>
+    <body>
+        <div class="content">
+            {% for text in texts %}
+                <p>{{ text }}</p>
+            {% endfor %}
+        </div>
+    </body>
+    </html>
+    """
+    template = Template(html_content)
+    rendered_html = template.render(texts=texts)
+    try:
+        # Convert HTML to PDF
+        pdfkit.from_string(rendered_html, 'result.pdf')
+        st.success("PDF файл 'result.pdf' успешно создан")
+    except Exception as e:
+        st.error(f"Ошибка при создании PDF: {str(e)}")
+        st.warning("PDF generation requires wkhtmltopdf to be installed")
 # Initialize sentiment analyzers
         llm = ChatOpenAI(
             base_url="https://api.groq.com/openai/v1",
             model="llama-3.1-70b-versatile",
+            openai_api_key=groq_api_key,  # Updated parameter name
             temperature=0.0
         )
         return llm
         unsafe_allow_html=True
     )
+    st.title("::: анализ мониторинга новостей СКАН-ИНТЕРФАКС (v.3.4):::")
     if 'processed_df' not in st.session_state:
         st.session_state.processed_df = None

requirements.txt CHANGED Viewed

@@ -16,4 +16,5 @@ huggingface_hub
 accelerate>=0.26.0
 openai
 wordcloud
-fpdf2 == 2.7.8

 accelerate>=0.26.0
 openai
 wordcloud
+pdfkit
+Jinja2==3.1.2