Update app.py

app.py

@@ -1,13 +1,19 @@
-
-from user_utils import *
-
 import streamlit as st
 from dotenv import load_dotenv
+from transformers import pipeline
 
-def main():
-    # Load environment variables
-    load_dotenv()
+# Load environment variables
+load_dotenv()
 
+# Initialize the question-answering pipeline with a pre-trained model
+qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
+
+def fine_tune_extraction(content, query):
+    # Use the pipeline to answer the question using the document content
+    result = qa_pipeline(question=query, context=content)
+    return result['answer']
+
+def main():
     st.header("Automatic Ticket Classification Tool")
     
     # Capture user input
@@ -16,7 +22,7 @@ def main():
 
     if user_input:
         try:
-            # Create embeddings instance
+            # Assuming create_embeddings and pull_from_pinecone are defined elsewhere
             embeddings = create_embeddings()
 
             # Fetch the Pinecone index using the API key and environment info
@@ -37,7 +43,7 @@ def main():
                     # Access the document content directly
                     content = getattr(doc, "page_content", "No content available.")  # Safely access content
 
-                    # Apply fine-tuned extraction model
+                    # Apply fine-tuned extraction model to extract relevant information
                     relevant_info = fine_tune_extraction(content, user_input)
 
                     if relevant_info: