Update app.py

app.py

@@ -6,8 +6,6 @@ import os
 import numpy as np
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
-import html
-from docx import Document  # New import for working with docx files
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(pdf_file):
@@ -28,23 +26,6 @@ def search_similar(query_embedding, index, stored_texts, top_k=3):
     results = [(stored_texts[i], distances[0][idx]) for idx, i in enumerate(indices[0])]
     return results
 
-# Function to generate a docx file with the response content
-def generate_docx(response_content):
-    doc = Document()
-    
-    # Adding title and response to the docx
-    doc.add_heading('Course Query Response', 0)
-    doc.add_heading('Answer:', level=1)
-    doc.add_paragraph(response_content)
-    
-    # Save the document to a byte stream to allow for download in Streamlit
-    from io import BytesIO
-    doc_io = BytesIO()
-    doc.save(doc_io)
-    doc_io.seek(0)
-    
-    return doc_io
-
 # Streamlit app starts here
 st.title("Course Query Assistant")
 
@@ -100,7 +81,7 @@ if openai_api_key:
 
             # Get the GPT-4 response
             response = openai.ChatCompletion.create(
-                model="gpt-4o-mini",  # Update to GPT-4 (or your desired model)
+                model="gpt-4",  # Update to GPT-4 (or your desired model)
                 messages=[{"role": "user", "content": modified_prompt}]
             )
 
@@ -110,14 +91,3 @@ if openai_api_key:
             # Display the response in Streamlit (Intelligent Reply)
             st.write("### Intelligent Reply:")
             st.write(response_content)
-
-            # Generate a docx file based on the exact response content
-            docx_file = generate_docx(response_content)
-
-            # Provide the download button for the .docx file
-            st.download_button(
-                label="Download Response as DOCX",
-                data=docx_file,
-                file_name="course_query_response.docx",
-                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-            )