Update app.py

app.py

@@ -3,127 +3,88 @@ from PyPDF2 import PdfReader
 import requests
 import re
 
-# Configure page
+# Set page config first (only needed once)
 st.set_page_config(
-    page_title="Perfect Study Assistant",
+    page_title="Smart Study Assistant",
     page_icon="🎓",
     layout="wide"
 )
 
+# Initialize session state (if needed)
+if 'generated' not in st.session_state:
+    st.session_state.generated = False
+
 def extract_text(pdf_file):
-    """Clean PDF text extraction"""
-    text = ""
+    """Simplified PDF text extraction"""
+    if not pdf_file:
+        st.warning("Please upload a PDF file")
+        return None
     try:
-        reader = PdfReader(pdf_file)
-        for page in reader.pages:
-            text += page.extract_text() + "\n\n"
-        return re.sub(r'\n{3,}', '\n\n', text)[:3000]  # Remove excessive newlines
+        return "\n\n".join([page.extract_text() for page in PdfReader(pdf_file).pages])[:3000]
     except Exception as e:
         st.error(f"PDF Error: {str(e)}")
         return None
 
 def call_ai(prompt):
-    """Optimized AI call with strict output control"""
+    """Streamlined API call"""
     try:
         response = requests.post(
             "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
             headers={"Authorization": f"Bearer {st.secrets.HF_TOKEN}"},
-            json={
-                "inputs": prompt,
-                "parameters": {
-                    "temperature": 0.3,  # Less creative but more focused
-                    "max_new_tokens": 800,
-                    "do_sample": False  # More deterministic
-                }
-            },
-            timeout=45
+            json={"inputs": prompt},
+            timeout=30
         )
-        if response.status_code == 200:
-            return clean_output(response.json()[0]['generated_text'])
-        return f"⚠️ Error: {response.text[:200]}"
+        return response.json()[0]['generated_text'] if response.status_code == 200 else f"Error: {response.text}"
     except Exception as e:
-        return f"⚠️ Connection Error: {str(e)}"
-
-def clean_output(text):
-    """Remove all AI instructions and formatting artifacts"""
-    text = re.sub(r'(?:Create|Generate).*?:', '', text)  # Remove prompt prefixes
-    text = re.sub(r'Page \d+.*?\n', '', text)  # Remove page numbers
-    text = re.sub(r'©.*?\n', '', text)  # Remove copyrights
-    text = re.sub(r'\n{3,}', '\n\n', text)  # Normalize newlines
-    return text.strip()
+        return f"Connection Error: {str(e)}"
 
 def generate_content(text):
-    """Precise content generation with strict templates"""
+    """Direct content generation without separate materials dict"""
     return {
-        "summary": call_ai(f"""Extract ONLY key concepts as bullet points:
-                        - Concept 1
-                        - Concept 2
-                        - Sub-concept 2.1
-                        From this text (ONLY include the bullet points, no intro/outro):
-                        {text}"""),
-        
-        "mindmap": call_ai(f"""Create Mermaid mindmap syntax (ONLY the code):
-                        ```mermaid
-                        mindmap
-                          root((Core Concept))
-                            Main Idea 1
-                              Detail 1
-                            Main Idea 2
-                        ```
-                        For:
-                        {text}"""),
-        
-        "flashcards": call_ai(f"""Generate EXACTLY 5 flashcards (ONLY Q/A pairs):
-                        Q1: Clear question?
-                        A1: Concise answer
-                        ---
-                        Q2: Next question?
-                        A2: Answer
-                        From:
-                        {text}"""),
-        
-        "mcq": call_ai(f"""Create EXACTLY 5 MCQs (ONLY questions/answers):
-                        1. Question?
-                        a) Option 1
-                        b) Option 2
-                        c) Option 3
-                        d) Option 4
-                        Answer: a
-                        From:
-                        {text}""")
+        "summary": call_ai(f"Create structured markdown summary:\n{text}"),
+        "bullets": call_ai(f"Convert to hierarchical bullets:\n{text}"),
+        "mindmap": call_ai(f"Generate Mermaid mindmap syntax:\n{text}"),
+        "flashcards": call_ai(f"Create 5 flashcards:\n{text}"),
+        "mcq": call_ai(f"Generate 5 MCQs:\n{text}")
     }
 
-# Streamlit UI
-st.title("🎯 Perfect Study Assistant")
-uploaded_file = st.file_uploader("Upload PDF (2-3 pages work best)", type="pdf")
+# UI Components
+st.title("🎓 Smart Study Assistant")
+uploaded_file = st.file_uploader("Upload PDF Lecture Notes", type="pdf")
 
-if uploaded_file and st.button("Generate Precise Materials"):
-    with st.spinner("Creating perfect study materials..."):
+if uploaded_file:
+    if st.button("Generate Study Materials") or st.session_state.generated:
+        st.session_state.generated = True
         text = extract_text(uploaded_file)
+        
         if text and len(text) > 100:
-            content = generate_content(text)
-            
-            col1, col2 = st.columns(2)
-            
-            with col1:
-                with st.expander("📝 Clean Summary", expanded=True):
+            with st.spinner("Generating materials..."):
+                content = generate_content(text)
+                
+                # Display results in tabs
+                tab1, tab2, tab3, tab4, tab5 = st.tabs([
+                    "Summary", "Bullet Points", "Mind Map", 
+                    "Flashcards", "MCQ Quiz"
+                ])
+                
+                with tab1:
                     st.markdown(content["summary"])
                 
-                with st.expander("📇 Flashcards"):
+                with tab2:
+                    st.markdown(content["bullets"])
+                
+                with tab3:
+                    st.code(content["mindmap"], language="mermaid")
+                    st.components.v1.html(f"""
+                    <div class="mermaid">{content['mindmap']}</div>
+                    <script src="https://cdn.jsdelivr.net/npm/mermaid/dist/mermaid.min.js"></script>
+                    <script>mermaid.initialize({{startOnLoad:true}});</script>
+                    """, height=400)
+                
+                with tab4:
                     st.markdown(content["flashcards"])
                 
-                with st.expander("❓ MCQ Quiz"):
+                with tab5:
                     st.markdown(content["mcq"])
-            
-            with col2:
-                st.subheader("🗺️ Interactive Mind Map")
-                st.code(content["mindmap"], language="mermaid")
-                st.components.v1.html(f"""
-                <div class="mermaid">
-                {content['mindmap'].replace('```mermaid', '').replace('```', '')}
-                </div>
-                <script src="https://cdn.jsdelivr.net/npm/mermaid/dist/mermaid.min.js"></script>
-                <script>mermaid.initialize({{startOnLoad:true}});</script>
-                """, height=500)
         else:
-            st.warning("Text too short - try a more substantial PDF")
+            st.warning("PDF text too short - try another file")