Create app.py

app.py

@@ -0,0 +1,109 @@
+import streamlit as st
+import torch
+from audio_handler import AudioHandler
+from chat_handler import ChatHandler
+from transformers import pipeline
+import base64
+import time
+
+# Page configuration
+st.set_page_config(
+    page_title="Darija Mental Health Assistant",
+    page_icon="🤗",
+    layout="centered"
+)
+
+# Custom CSS
+st.markdown("""
+<style>
+    .stButton > button {
+        background-color: #4CAF50;
+        color: white;
+        padding: 10px 20px;
+        border-radius: 5px;
+    }
+    .main-header {
+        text-align: center;
+        color: #2e7d32;
+    }
+    .chat-container {
+        padding: 20px;
+        border-radius: 10px;
+        background-color: #f5f5f5;
+        margin: 10px 0;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# Initialize session state
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+if 'recording' not in st.session_state:
+    st.session_state.recording = False
+if 'audio_handler' not in st.session_state:
+    st.session_state.audio_handler = AudioHandler()
+if 'chat_handler' not in st.session_state:
+    st.session_state.chat_handler = ChatHandler()
+
+def main():
+    # Header
+    st.markdown("<h1 class='main-header'>Darija Mental Health Assistant 🤗</h1>", unsafe_allow_html=True)
+    st.markdown("<h3 style='text-align: center;'>تحدث معي بالدارجة عن مشاعرك</h3>", unsafe_allow_html=True)
+
+    # Sidebar for settings
+    with st.sidebar:
+        st.header("Settings ⚙️")
+        voice_enabled = st.toggle("Enable Voice Input 🎤", True)
+        st.divider()
+        st.markdown("### About")
+        st.info("This assistant provides mental health support in Moroccan Arabic (Darija). Feel free to speak or type your thoughts.")
+
+    # Main chat interface
+    st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
+    
+    # Voice input section
+    if voice_enabled:
+        cols = st.columns([1, 1])
+        with cols[0]:
+            if st.button("🎤 Start Recording", disabled=st.session_state.recording):
+                st.session_state.recording = True
+                st.session_state.audio_handler.start_recording()
+                st.experimental_rerun()
+        
+        with cols[1]:
+            if st.button("⏹️ Stop Recording", disabled=not st.session_state.recording):
+                if st.session_state.recording:
+                    audio_file = st.session_state.audio_handler.stop_recording()
+                    with st.spinner("Processing your message..."):
+                        transcription = st.session_state.audio_handler.transcribe_audio(audio_file)
+                        process_input(transcription)
+                    st.session_state.recording = False
+                    st.experimental_rerun()
+
+    # Text input
+    user_input = st.text_input("اكتب رسالتك هنا:", key="text_input")
+    if user_input:
+        process_input(user_input)
+
+    # Display chat history
+    display_chat_history()
+
+    st.markdown("</div>", unsafe_allow_html=True)
+
+def process_input(user_input):
+    if user_input:
+        # Add user message
+        st.session_state.messages.append({"role": "user", "content": user_input})
+        
+        # Get bot response
+        with st.spinner("جاري التفكير..."):
+            response = st.session_state.chat_handler.get_response(user_input)
+            st.session_state.messages.append({"role": "assistant", "content": response})
+
+def display_chat_history():
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.write(message["content"])
+
+if __name__ == "__main__":
+    main()