Update app.py

app.py

@@ -0,0 +1,179 @@
+import streamlit as st
+import json
+import asyncio
+import edge_tts
+from huggingface_hub import InferenceClient
+
+# Configuration
+EXTRACTOR_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
+PERSONALITY_MODEL = "HuggingFaceH4/zephyr-7b-beta"
+
+# Synthetic Data: 30 Messages reflecting personality traits (Big Five / MBTI context)
+DEFAULT_CHAT_LOG = """
+1. User: I feel completely drained after large social gatherings. I need alone time to recharge.
+2. User: I enjoy abstract theories more than practical details.
+3. User: My desk is always messy, but I know where everything is.
+4. User: I often worry that I said the wrong thing in conversations.
+5. User: I prefer planning everything weeks in advance. Spontaneity stresses me out.
+6. User: I find it hard to sympathize with people who are overly emotional.
+7. User: I love exploring new ideas and hobbies, even if I don't finish them.
+8. User: Criticism hits me really hard, even if it's constructive.
+9. User: I tend to take charge in group projects because I want things done right.
+10. User: I value logic and truth over sparing someone's feelings.
+11. User: I often get lost in my own thoughts and forget my surroundings.
+12. User: I hate conflict. I will do anything to keep the peace.
+13. User: I feel a strong need to help others, sometimes at my own expense.
+14. User: I get bored easily with routine tasks.
+15. User: I am very skeptical until I see concrete evidence.
+16. User: I love meeting new people and being the center of attention.
+17. User: I struggle to express my emotions verbally.
+18. User: I often procrastinate until the very last minute.
+19. User: I feel deeply moved by art and music.
+20. User: I prefer a few close friends over a large group of acquaintances.
+21. User: I have a hard time saying "no" to people.
+22. User: I am constantly analyzing why people do what they do.
+23. User: I value tradition and established rules.
+24. User: I am often described as "intense" or "serious."
+25. User: I feel energetic when I'm debating intellectual topics.
+26. User: I worry a lot about the future.
+27. User: I trust my gut instincts more than data sometimes.
+28. User: I prefer to work alone rather than in a team.
+29. User: I am very competitive and hate losing.
+30. User: I want to understand my true purpose in life.
+"""
+
+st.set_page_config(page_title="Personality Analysis Engine", layout="wide")
+
+# Authentication Sidebar
+with st.sidebar:
+    st.header("Configuration")
+    hf_token = st.text_input("Hugging Face Token", type="password")
+    if not hf_token:
+        st.warning("Please enter your Hugging Face Token to proceed.")
+        st.stop()
+
+client = InferenceClient(token=hf_token)
+
+# Module A: Memory Extraction
+def extract_memory(chat_logs):
+    # System prompt designed to extract psychological profile
+    system_prompt = """
+    You are a Psychological Analyst. Analyze the chat logs and extract a structured user profile.
+    
+    Required JSON Keys:
+    1. "traits": Big Five or MBTI indicators (e.g., Introversion, High Openness, Agreeableness).
+    2. "values": What the user prioritizes (e.g., Logic, Harmony, Efficiency, Autonomy).
+    3. "struggles": Recurring points of friction (e.g., Social anxiety, Procrastination, Conflict avoidance).
+    
+    Output strictly valid JSON only.
+    """
+    
+    prompt = f"<|system|>\n{system_prompt}</s>\n<|user|>\n{chat_logs}</s>\n<|assistant|>"
+    
+    try:
+        with st.spinner("Analyzing psychological patterns..."):
+            response = client.text_generation(
+                model=EXTRACTOR_MODEL,
+                prompt=prompt,
+                max_new_tokens=600,
+                temperature=0.1
+            )
+            # Parse JSON from response
+            text = response.strip()
+            start = text.find("{")
+            end = text.rfind("}") + 1
+            if start != -1 and end != -1:
+                return json.loads(text[start:end])
+            else:
+                return {"error": "Failed to parse JSON output"}
+    except Exception as e:
+        return {"error": str(e)}
+
+# Module B: Personality Engine (Text Generation)
+def generate_response(message, memory, persona):
+    # Define distinct psychological personas
+    personas = {
+        "Analytical Psychologist": "Role: Jungian Analyst. Tone: Objective, deep, intellectual. Focus: Unconscious patterns, archetypes, and logic.",
+        "Empathetic Counselor": "Role: Humanistic Therapist. Tone: Warm, validating, gentle. Focus: Emotional safety, acceptance, and feelings.",
+        "Pragmatic Coach": "Role: Behavioral Coach. Tone: Direct, action-oriented, blunt. Focus: Solutions, efficiency, and breaking habits."
+    }
+    
+    context = f"""
+    USER PSYCH PROFILE:
+    - Traits: {memory.get('traits', 'Unknown')}
+    - Values: {memory.get('values', 'Unknown')}
+    - Struggles: {memory.get('struggles', 'Unknown')}
+    """
+    
+    messages = [
+        {"role": "system", "content": f"{personas[persona]}\n\n{context}"},
+        {"role": "user", "content": message}
+    ]
+    
+    try:
+        res = client.chat_completion(
+            model=PERSONALITY_MODEL,
+            messages=messages,
+            max_tokens=250,
+            temperature=0.7
+        )
+        return res.choices[0].message.content
+    except Exception as e:
+        return f"Error generating text: {e}"
+
+# Module C: Audio Engine (Edge-TTS)
+async def generate_audio(text, persona):
+    # Map personas to suitable voice types
+    voice_map = {
+        "Analytical Psychologist": "en-US-ChristopherNeural", # Deep, serious male voice
+        "Empathetic Counselor": "en-US-AvaNeural",            # Soft, soothing female voice
+        "Pragmatic Coach": "en-US-EricNeural"                 # Energetic, firm male voice
+    }
+    
+    voice = voice_map.get(persona, "en-US-AriaNeural")
+    output_file = "response.mp3"
+    
+    communicate = edge_tts.Communicate(text, voice)
+    await communicate.save(output_file)
+    return output_file
+
+# Main UI Layout
+st.title("Human Personality Engine")
+
+col1, col2 = st.columns([1, 1])
+
+# Left Column: Data Analysis
+with col1:
+    st.subheader("1. Profile Extraction")
+    with st.expander("View Raw Chat Logs"):
+        st.text(DEFAULT_CHAT_LOG)
+        
+    if st.button("Analyze User Profile"):
+        memory_data = extract_memory(DEFAULT_CHAT_LOG)
+        st.session_state['memory'] = memory_data
+        st.success("Analysis Complete")
+    
+    if 'memory' in st.session_state:
+        st.json(st.session_state['memory'])
+
+# Right Column: Interaction
+with col2:
+    st.subheader("2. Agent Interaction")
+    
+    if 'memory' not in st.session_state:
+        st.info("Please analyze the user profile first.")
+    else:
+        user_input = st.text_input("Input Message:", "I feel overwhelmed by all the details.")
+        persona = st.selectbox("Select Agent Persona:", ["Analytical Psychologist", "Empathetic Counselor", "Pragmatic Coach"])
+        
+        if st.button("Generate Response"):
+            # Generate Text
+            with st.spinner("Generating text response..."):
+                reply = generate_response(user_input, st.session_state['memory'], persona)
+            st.markdown(f"**{persona}:** {reply}")
+            
+            # Generate Audio
+            with st.spinner("Synthesizing speech..."):
+                asyncio.run(generate_audio(reply, persona))
+            
+            st.audio("response.mp3")