Update app.py

app.py

@@ -7,8 +7,6 @@ from io import BytesIO
 import os
 from dotenv import load_dotenv
 from groq import Groq
-import requests
-from bs4 import BeautifulSoup
 
 # Load API key
 load_dotenv()
@@ -21,7 +19,7 @@ except Exception as e:
     st.error(f"Failed to initialize Groq client: {str(e)}")
     client = None
 
-# Extract resume text
+# Extract text from uploaded file
 def extract_text(file):
     if file.name.endswith(".pdf"):
         try:
@@ -36,34 +34,25 @@ def extract_text(file):
             st.error(f"DOCX error: {e}")
     return None
 
-# Extract job description from URL
-def extract_from_url(url):
-    try:
-        response = requests.get(url, headers={"User-Agent": "Mozilla"}, timeout=10)
-        soup = BeautifulSoup(response.text, "html.parser")
-        for tag in soup(["script", "style"]):
-            tag.decompose()
-        return " ".join(soup.stripped_strings)
-    except Exception as e:
-        st.error(f"URL extract error: {e}")
-        return None
-
-# AI using llama3 (replaces broken deepseek)
+# Analyze with Groq API
 def analyze_with_ai(prompt):
     if not client:
         return "Missing API key"
     try:
-        response = client.chat.completions.create(
+        completion = client.chat.completions.create(
             model="llama3-70b-8192",
             messages=[{"role": "user", "content": prompt}],
-            temperature=0.3,
-            max_tokens=1800,
+            temperature=0.7,  # Balanced creativity for feedback
+            max_tokens=4096,  # Adjusted for detailed feedback
+            top_p=1,
+            stream=False,
+            stop=None
         )
-        return response.choices[0].message.content
+        return completion.choices[0].message.content
     except Exception as e:
         return f"AI Error: {e}"
 
-# Save text as docx
+# Save text as DOCX
 def text_to_docx(text):
     doc = Document()
     for p in text.split("\n"):
@@ -74,82 +63,89 @@ def text_to_docx(text):
     buffer.seek(0)
     return buffer
 
-# UI
-st.set_page_config(page_title="AI Resume & Cover Letter Assistant", layout="wide")
-st.title("🤖 AI Resume & Cover Letter Assistant")
-st.write("Upload your resume and job description to generate a custom resume + cover letter.")
+# Streamlit UI
+st.set_page_config(page_title="AI IELTS Preparation Assistant", layout="wide")
+st.title("📚 AI IELTS Preparation Assistant")
+st.write("Upload your IELTS essay or input text to get feedback and practice questions for Writing and Speaking.")
 
 with st.sidebar:
     st.markdown("### Instructions")
-    st.markdown("1. Upload your resume (PDF or DOCX)")
-    st.markdown("2. Paste a job description or job URL")
-    st.markdown("3. Click 'Analyze'")
-    st.markdown("4. Download your optimized resume and cover letter")
+    st.markdown("1. Upload an essay (PDF or DOCX) or enter text.")
+    st.markdown("2. Select the IELTS section (Writing/Speaking).")
+    st.markdown("3. Click 'Analyze' for feedback and practice questions.")
+    st.markdown("4. Download feedback and practice materials.")
 
 # Inputs
 col1, col2 = st.columns(2)
 with col1:
-    resume_file = st.file_uploader("Upload Resume", type=["pdf", "docx"])
+    essay_file = st.file_uploader("Upload Essay (Optional)", type=["pdf", "docx"])
 with col2:
-    job_input = st.text_input("Job Description (URL or Text)")
+    essay_text = st.text_area("Or Enter Essay Text", height=200)
+    ielts_section = st.selectbox("IELTS Section", ["Writing Task 1", "Writing Task 2", "Speaking"])
 
 if st.button("🔍 Analyze and Generate"):
-    if resume_file and job_input:
-        with st.spinner("Processing your resume and job info..."):
-            resume_text = extract_text(resume_file)
-            job_text = extract_from_url(job_input) if job_input.startswith("http") else job_input
-
-            if not resume_text or not job_text:
-                st.error("Resume or job description could not be processed.")
+    if essay_file or essay_text:
+        with st.spinner("Analyzing your input..."):
+            # Extract text from file or use direct input
+            input_text = extract_text(essay_file) if essay_file else essay_text
+            if not input_text:
+                st.error("No text provided or extraction failed.")
                 st.stop()
 
-            prompt = f"""
-Given the following resume and job description, improve the resume by injecting relevant ATS keywords and tailoring it to the job. Also generate a custom cover letter.
-
-RESUME:
-{resume_text[:3000]}
-
-JOB DESCRIPTION:
-{job_text[:3000]}
-
-Respond in this format:
-### Optimized Resume:
-<Improved resume>
-
-### Cover Letter:
-<Professional cover letter>
+            # Define prompt based on IELTS section
+            if ielts_section.startswith("Writing"):
+                prompt = f"""
+You are an IELTS examiner. Analyze the following {ielts_section} essay for grammar, vocabulary, coherence, and task response. Provide detailed feedback, including strengths, weaknesses, and suggestions for improvement (aligned with IELTS band descriptors). Also, generate 3 practice questions for {ielts_section}.  
+ESSAY:  
+{input_text[:3000]}  
+Respond in this format:  
+### Feedback:  
+<Detailed feedback>  
+### Practice Questions:  
+<Three practice questions>  
+"""
+            else:  # Speaking
+                prompt = f"""
+You are an IELTS examiner. Analyze the following IELTS Speaking response for fluency, vocabulary, grammar, and pronunciation. Provide detailed feedback, including strengths, weaknesses, and suggestions for improvement (aligned with IELTS band descriptors). Also, generate 3 practice questions for IELTS Speaking.  
+RESPONSE:  
+{input_text[:3000]}  
+Respond in this format:  
+### Feedback:  
+<Detailed feedback>  
+### Practice Questions:  
+<Three practice questions>  
 """
 
             ai_response = analyze_with_ai(prompt)
 
-            if "### Optimized Resume:" in ai_response and "### Cover Letter:" in ai_response:
-                resume_part = ai_response.split("### Cover Letter:")[0].replace("### Optimized Resume:", "").strip()
-                cover_letter_part = ai_response.split("### Cover Letter:")[1].strip()
+            if "### Feedback:" in ai_response and "### Practice Questions:" in ai_response:
+                feedback_part = ai_response.split("### Practice Questions:")[0].replace("### Feedback:", "").strip()
+                questions_part = ai_response.split("### Practice Questions:")[1].strip()
             else:
-                resume_part = ai_response.strip()
-                cover_letter_part = ""
-                st.warning("Unexpected format. Showing full response as resume.")
-
-            # Show & Download Resume
-            st.success("✅ Documents generated!")
-            st.subheader("📄 Optimized Resume")
-            st.markdown(resume_part)
-            st.download_button("⬇ Download Resume", data=text_to_docx(resume_part), file_name="optimized_resume.docx")
-
-            # Show & Download Cover Letter
-            if cover_letter_part:
-                st.subheader("📄 Cover Letter")
-                st.markdown(cover_letter_part)
-                st.download_button("⬇ Download Cover Letter", data=text_to_docx(cover_letter_part), file_name="cover_letter.docx")
+                feedback_part = ai_response.strip()
+                questions_part = ""
+                st.warning("Unexpected response format. Showing full response as feedback.")
+
+            # Display and download feedback
+            st.success("✅ Analysis complete!")
+            st.subheader("📝 Feedback")
+            st.markdown(feedback_part)
+            st.download_button("⬇ Download Feedback", data=text_to_docx(feedback_part), file_name="ielts_feedback.docx")
+
+            # Display and download practice questions
+            if questions_part:
+                st.subheader("❓ Practice Questions")
+                st.markdown(questions_part)
+                st.download_button("⬇ Download Practice Questions", data=text_to_docx(questions_part), file_name="practice_questions.docx")
             else:
-                st.info("No cover letter was generated.")
+                st.info("No practice questions generated.")
     else:
-        st.warning("Please upload both resume and job input.")
+        st.warning("Please upload an essay or enter text.")
 
 # Footer
 st.markdown("---")
 st.markdown("""
 <div style='text-align:center; font-size:0.85em; color:gray'>
-    AI Career Assistant · Powered by Groq LLaMA3 · Built with ❤️ in Streamlit
+    AI IELTS Assistant · Powered by Groq LLaMA3 · Built with ❤️ in Streamlit
 </div>
-""", unsafe_allow_html=True)
+""", unsafe_allow_html=True)