Update app.py

app.py

@@ -1,27 +1,28 @@
+# app.py
 import streamlit as st
-from utils import save_uploaded_audio
-from voice_cloner import clone_and_generate_text
 import os
 import whisper
-import torchaudio
+from groq_llm import get_groq_response
+from utils import save_uploaded_audio, transcribe_audio, save_chat_history, list_chat_histories, load_chat_history
+from voice_cloner import clone_and_generate_text
 from groq import Groq
-from utils import save_chat_history, load_chat_history, list_chat_histories
-
 
-
-# Load Whisper model for transcription
+# Load Whisper model
 whisper_model = whisper.load_model("tiny")
 
-# Initialize Groq LLM client
+# Initialize Groq client
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 st.set_page_config(page_title="Voice Chat", layout="centered")
 st.title("🎤💬 Voice & Text Chat using Your Cloned Voice")
 
-# Store cloned voice path
 if "clone_path" not in st.session_state:
     st.session_state.clone_path = None
 
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
+# Sidebar: voice setup
 st.sidebar.header("🧬 Setup Your Clone Voice")
 voice_option = st.sidebar.radio("Choose how to provide clone voice", ["Upload Voice"])
 
@@ -32,34 +33,27 @@ if voice_option == "Upload Voice":
         st.session_state.clone_path = path
         st.success("✅ Voice uploaded and saved as your clone voice.")
 
-# --- Conversation section ---
+# Tabs for input
 st.subheader("🗣️ Ask with voice or type text below")
-
 tab1, tab2 = st.tabs(["🎤 Voice Input", "💬 Text Input"])
 
-# --- VOICE INPUT TAB ---
+# Voice input tab
 with tab1:
     user_voice = st.file_uploader("Upload your voice question", type=["wav", "mp3", "m4a", "flac", "ogg"])
     if user_voice:
         user_voice_path = save_uploaded_audio(user_voice, "user_question.wav")
         st.audio(user_voice_path)
 
-        # Step 1: Transcribe voice
         st.info("Transcribing...")
-        result = whisper_model.transcribe(user_voice_path)
-        user_text = result["text"]
+        user_text = transcribe_audio(user_voice_path, language="Urdu")
         st.success(f"📝 You said: {user_text}")
 
-        # Step 2: Get LLM response
         st.info("Thinking...")
-        response = groq_client.chat.completions.create(
-            model="llama3-8b-8192",
-            messages=[{"role": "user", "content": user_text}]
-        )
-        reply = response.choices[0].message.content
+        reply = get_groq_response(user_text, system_prompt="آپ ایک مددگار AI ہیں، جو اردو میں جواب دیتی ہے", language="Urdu")
         st.success(f"🤖 AI says: {reply}")
 
-        # Step 3: Voice reply
+        st.session_state.chat_history.append({"user": user_text, "ai": reply})
+
         if st.session_state.clone_path:
             st.info("Cloning voice reply...")
             voice_output_path = clone_and_generate_text(reply, st.session_state.clone_path)
@@ -67,32 +61,38 @@ with tab1:
         else:
             st.warning("Upload your voice clone first in the sidebar.")
 
-# --- TEXT INPUT TAB ---
+# Text input tab
 with tab2:
     user_input = st.text_input("Type your question here:")
     if st.button("Send Text"):
         if user_input.strip() == "":
             st.warning("Please enter a message.")
         else:
-            # Step 1: Get LLM response
             st.info("Thinking...")
-            response = groq_client.chat.completions.create(
-                model="llama3-8b-8192",
-                messages=[{"role": "user", "content": user_input}]
-            )
-            reply = response.choices[0].message.content
+            reply = get_groq_response(user_input, system_prompt="You are a helpful AI.", language="Urdu")
             st.success(f"🤖 AI says: {reply}")
 
-            # Step 2: Voice reply
+            st.session_state.chat_history.append({"user": user_input, "ai": reply})
+
             if st.session_state.clone_path:
                 st.info("Cloning voice reply...")
                 voice_output_path = clone_and_generate_text(reply, st.session_state.clone_path)
                 st.audio(voice_output_path)
             else:
                 st.warning("Upload your voice clone first in the sidebar.")
-# Footer
-st.markdown("""
-    <div class='footer'>
-        DESIGN BY MERAJ GRAPHICS
-    </div>
-""", unsafe_allow_html=True)
+
+# Save history
+if st.button("Save Chat History"):
+    filename = save_chat_history(st.session_state.chat_history)
+    st.success(f"History saved as {filename}")
+
+# View past chat histories
+st.sidebar.markdown("---")
+st.sidebar.subheader("📃 Load Chat History")
+chat_files = list_chat_histories()
+selected_file = st.sidebar.selectbox("Select a file to load", chat_files)
+if st.sidebar.button("Load History"):
+    history = load_chat_history(selected_file)
+    for entry in history:
+        st.write(f"**You:** {entry['user']}")
+        st.write(f"**AI:** {entry['ai']}")