Aktualizacja pliku zależności i głównej aplikacji

app.py

@@ -7,6 +7,8 @@ from langchain_openai import ChatOpenAI
 from langchain_core.output_parsers import StrOutputParser
 from pydantic import BaseModel, Field, validator
 from pydantic import BaseModel, Field, field_validator
+from langchain_community.document_loaders import PyPDFLoader, Docx2txtLoader
+import os
 
 
 # %%
@@ -92,7 +94,27 @@ def prepare_questions(df):
 
 
 # %%
-def analyze_job_ad(job_ad):
+# Konwersja plików PDF i Word do tekstu
+def doc_to_text(file):
+    extension = os.path.splitext(file)[1].lower()
+    if extension==".docx":
+        loader = Docx2txtLoader(file)
+    elif extension==".pdf":
+        loader = PyPDFLoader(file)
+    else:
+        return "error"
+    pages = loader.load()
+    text=''
+    for page in pages:
+            text += page.page_content + "\n"
+    return text
+
+
+def analyze_job_ad(job_ad, file):
+    if file:
+        job_ad=doc_to_text(file)
+        if job_ad == "error":
+            return "{}", None
     """Analizuje ogłoszenie o pracę przy użyciu LangChain i OpenAI."""
     questions = prepare_questions(matryca_df)
     prompt_template = PromptTemplate.from_template(
@@ -155,7 +177,7 @@ from io import BytesIO
 from datetime import datetime
 import tempfile
 def create_report(result: pd.DataFrame) -> BytesIO:
-    doc = Document()
+    doc = Document('template.docx')
     doc.add_heading('Raport analizy ogłoszenia o pracę', 0)
     doc.add_paragraph(f'Data wygenerowania: {datetime.now().strftime("%d.%m.%Y %H:%M")}')
     for _, row in result.iterrows():
@@ -185,8 +207,8 @@ def create_report(result: pd.DataFrame) -> BytesIO:
 # %%
 demo=gr.Interface(
     fn=analyze_job_ad,
-    inputs=gr.TextArea(),
-    outputs=[gr.TextArea(), gr.DownloadButton(label='Pobierz raport w formacie Word')],
+    inputs=[gr.TextArea(), gr.File()],
+    outputs=[gr.JSON(), gr.DownloadButton(label='Pobierz raport w formacie Word')],
     title="KoREKtor"
 ).launch(inbrowser=True)
 

requirements.txt

@@ -4,4 +4,7 @@ langchain_core
 langchain_openai
 langchain
 pydantic
-openai
+openai
+langchain_community
+os
+docx