Update src/streamlit_app.py

removing cv upload input to text input

src/streamlit_app.py

@@ -1,14 +1,13 @@
 import streamlit as st
 import fitz  # PyMuPDF
 from dotenv import load_dotenv
-import io
 import os
 import requests
 
 # Load environment variables from .env file
 load_dotenv()
 
-# Check and set environment variables
+# Set environment variables for Azure OpenAI
 ENDPOINT = os.getenv("AZURE_OPENAI_ENDPOINT")
 API_KEY = os.getenv("AZURE_OPENAI_API_KEY")
 DEPLOYMENT_NAME = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")
@@ -30,18 +29,6 @@ headers = {
     "Authorization": f"Bearer {HuggingFace_API_KEY}"  # Replace with your actual API key
 }
 
-# Function to extract text from the uploaded PDF
-def extract_text_from_pdf(pdf_file):
-    # Read the uploaded file as a byte stream
-    pdf_bytes = pdf_file.read()
-
-    # Open the PDF from the byte stream
-    doc = fitz.open("pdf", pdf_bytes)  # Fix: use the correct format to open the byte stream
-    text = ""
-    for page in doc:
-        text += page.get_text()
-    return text
-
 # Function to extract relevant information from the CV using Hugging Face or Azure OpenAI
 def extract_info_from_openai(text):
     prompt = f"""
@@ -62,27 +49,24 @@ def extract_info_from_openai(text):
     # If the Hugging Face response is successful, extract the generated text
     if response.status_code == 200:
         result = response.json()  # Parse the JSON response
-        return result.get("generated_text", "Error: Unable to extract text.")
+        return result.get("generated_text", "Error: Unable to generate response.")
     else:
         return f"Error: {response.status_code} - {response.text}"
 
 # Streamlit App UI
 st.title("AI Screening")
-st.write("Upload a CV PDF file, and the app will extract relevant information such as job title, location, skills, experience, and education.")
-
-# File uploader
-uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
+st.write("Enter the CV text below, and the app will extract relevant information such as job title, location, skills, experience, and education.")
 
-if uploaded_file is not None:
-    # Extract text from the uploaded PDF
-    text = extract_text_from_pdf(uploaded_file)
+# Text area for entering CV content manually
+cv_text = st.text_area("Enter CV Text", height=300)
 
-    # Display the extracted text (optional)
-    st.subheader("Extracted Text from CV")
-    st.text_area("Text from CV", text, height=300)
+if cv_text:
+    # Display the entered CV text
+    st.subheader("Entered Text from CV")
+    st.text_area("CV Text", cv_text, height=300)
 
     # Extract relevant information using Hugging Face (or Azure OpenAI if you need)
-    extracted_info = extract_info_from_openai(text)
+    extracted_info = extract_info_from_openai(cv_text)
 
     # Display the extracted information
     st.subheader("Extracted Information")