Update app.py

app.py

@@ -24,56 +24,43 @@ model = AutoModelForCausalLM.from_pretrained(generator_name, trust_remote_code=T
 document_file = st.file_uploader("Húzd ide a dokumentumot vagy kattints a feltöltéshez", type=["pdf", "docx", "doc"])
 
 if document_file is not None:
-    # Dokumentum méretének ellenőrzése és szeletekre bontása
-    document_chunks = []
-    if document_file.type == "application/pdf":
-        with pdfplumber.open(document_file) as pdf:
-            for page in pdf.pages:
-                document_chunks.append(page.extract_text())
-    elif document_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
-        docx_file = docx.Document(document_file)
-        document_text = ""
-        for paragraph in docx_file.paragraphs:
-            document_chunks.append(paragraph.text)
-    elif document_file.type == "application/msword":
-        doc_file = docx.Document(document_file)
+    # Válaszgenerálás
+    if st.button("Generálj Választ"):
         document_text = ""
-        for paragraph in doc_file.paragraphs:
-            document_chunks.append(paragraph.text)
-    else:
-        st.error("A fájltípus nem támogatott. Kérlek válassz ki egy PDF, DOCX vagy DOC fájlt!")
 
-    # Előző beszélgetésekhez csatolható kontextus
-    context = st.text_area("Korábbi Beszélgetéshez Tartozó Kontextus", "")
+        if document_file.type == "application/pdf":
+            with pdfplumber.open(document_file) as pdf:
+                for page in pdf.pages:
+                    document_text += page.extract_text()
+        elif document_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+            docx_file = docx.Document(document_file)
+            for paragraph in docx_file.paragraphs:
+                document_text += paragraph.text
+        elif document_file.type == "application/msword":
+            doc_file = docx.Document(document_file)
+            for paragraph in doc_file.paragraphs:
+                document_text += paragraph.text
+        else:
+            st.error("A fájltípus nem támogatott. Kérlek válassz ki egy PDF, DOCX vagy DOC fájlt!")
 
-    # Kérdés mező hozzáadása
-    question = st.text_input("Kérdés a Dokumentumból", "")
-
-    # Válaszgenerálás
-    if st.button("Generálj Választ"):
-        generated_responses = []
-        for chunk in document_chunks:
-            if context:
-                input_text = f"{context} {chunk}"
-            else:
-                input_text = chunk
+        # Előző beszélgetésekhez csatolható kontextus
+        context = st.text_area("Korábbi Beszélgetéshez Tartozó Kontextus", "")
 
-            # Ha van kérdés, azt is hozzáadjuk a bemenethez
-            if question:
-                input_text += f" {question}"
+        # Kérdés mező hozzáadása
+        question = st.text_input("Kérdés a Dokumentumból", "")
 
-            # Model használata a válasz generálásához
-            response = generate_response(input_text, tokenizer, model)
-            generated_responses.append(response)
+        # Ha van kérdés, azt is hozzáadjuk a bemenethez
+        if question:
+            input_text = f"{context} {document_text} {question}" if context else f"{document_text} {question}"
+        else:
+            input_text = f"{context} {document_text}" if context else document_text
 
-        # Összefésüljük a válaszokat egyetlen szöveggé
-        final_response = " ".join(generated_responses)
+        # Model használata a válasz generálásához
+        response = generate_response(input_text, tokenizer, model)
 
         # Válasz megjelenítése
         st.subheader("Generált Válasz:")
-        st.write(final_response)
+        st.write(response)
 
         # Aktuális beszélgetés hozzáadása az előző beszélgetésekhez
-        st.session_state.previous_conversations.append({"input_text": input_text, "response": final_response})
-else:
-    st.warning("Kérlek válassz ki egy dokumentumfájlt!")
+        st.session_state.previous_conversations.append({"input_text": input_text, "response": response})