Update app.py

app.py

@@ -4,8 +4,6 @@ from st_audiorec import st_audiorec
 from gtts import gTTS
 import os
 
-
-
 # Constants
 TITLE = "AgriTure"
 DESCRIPTION = """
@@ -17,55 +15,46 @@ Hope this will be a Successful Project!!!
 """
 
 # Initialize client
-
-
-with st.sidebar:
-    system_promptSide = st.text_input("Optional system prompt:")
-    temperatureSide = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.9, step=0.05)
-    max_new_tokensSide = st.slider("Max new tokens", min_value=0.0, max_value=4096.0, value=4096.0, step=64.0)
-    ToppSide = st.slider("Top-p (nucleus sampling)", min_value=0.0, max_value=1.0, value=0.6, step=0.05)
-    RepetitionpenaltySide = st.slider("Repetition penalty", min_value=0.0, max_value=2.0, value=1.2, step=0.05)
-
 whisper_client = Client("https://sanchit-gandhi-whisper-large-v2.hf.space/")
 
+# Function to convert text to speech using gTTS
+def text_to_speech(text, lang='en'):
+    tts = gTTS(text=text, lang=lang, slow=False)
+    tts.save("response.mp3")
+    return "response.mp3"
 
+# Function to transcribe audio
 def transcribe(wav_path):
-    
     return whisper_client.predict(
-				wav_path,	# str (filepath or URL to file) in 'inputs' Audio component
-				"transcribe",	# str in 'Task' Radio component
-				api_name="/predict"
+        wav_path,
+        "transcribe",
+        api_name="/predict"
     )
 
 # Prediction function
-def predict(message, system_prompt='Your name is AgriaTure. You are developed by Achyuth. Empower yourself farming future with cutting-edge technology and sustainable practices. You need to cultivate a greener and more productive. You need to give short answer like 2 to 5 sentences. If the user asks for any information about subject, then answer with long paragraphs.', temperature=0.7, max_new_tokens=4096,Topp=0.5,Repetitionpenalty=1.2):
+def predict(message, system_prompt='Your name is AgriaTure. You are developed by Achyuth. Your developer is a 14-year old boy, who is intrested on AI and Machine Learning. Dont reveal your developers age, until the user asks for your developers age. When the user asks you for information, never exclaim the words, and directly come to point and answer to the user. You need to provide short text-answers, meaningful answers and directly to the point. Never miscalculate the calculations. You need to check and verify the answer and then provide to user, the correct answer. You need to provide accurate and correct information. If the user conveys with you in any language, you also should convey in the same language. For example, the user said, "నమస్తే", you need to reply "నమస్తే! నేను మీ అగ్రిట్యూర్‌ని. ఈ రోజు మీకు సహాయం చేయడానికి నేను ఇక్కడ ఉన్నాను. ఈ రోజు విషయం ఏమిటి? ఏదైనా విషయంలో ఏమైనా సందేహాలు ఉన్నాయా?". You need to definitely reply in the same language!', temperature=0.5, max_new_tokens=2048, Topp=0.5, Repetitionpenalty=1.2):
     with st.status("Starting client"):
         client = Client("https://huggingface-projects-llama-2-7b-chat.hf.space/")
         st.write("Requesting Audio Transcriber")
     with st.status("Requesting AgriTure v1"):
         st.write("Requesting API")
         response = client.predict(
-    			message,	# str in 'Message' Textbox component
-                system_prompt,	# str in 'Optional system prompt' Textbox component
-    			max_new_tokens,	# int | float (numeric value between 0 and 4096)
-    			temperature,	# int | float (numeric value between 0.0 and 1.0)
-    			Topp,	
-                500,
-    			Repetitionpenalty,	# int | float (numeric value between 1.0 and 2.0)
-    			api_name="/chat"
+            message,
+            system_prompt,
+            max_new_tokens,
+            temperature,
+            Topp,
+            500,
+            Repetitionpenalty,
+            api_name="/chat"
         )
         st.write("Done")
         return response
 
-    
 # Streamlit UI
 st.title(TITLE)
 st.write(DESCRIPTION)
 
-
-
-
-
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
@@ -77,36 +66,50 @@ for message in st.session_state.messages:
 textinput = st.chat_input("Ask AgriTure anything...")
 wav_audio_data = st_audiorec()
 
-if wav_audio_data != None:
+if wav_audio_data is not None:
     with st.status("Transcribing audio..."):
         # save audio
         with open("audio.wav", "wb") as f:
             f.write(wav_audio_data)
         prompt = transcribe("audio.wav")
-       
         st.write("Transcribed Given Audio ✔")
-        
-    st.chat_message("human",avatar = "🌿").markdown(prompt)
+
+    st.chat_message("human", avatar="🌿").markdown(prompt)
     st.session_state.messages.append({"role": "human", "content": prompt})
 
     # transcribe audio
-    response = predict(message= prompt)
+    response = predict(message=prompt)
 
     with st.chat_message("assistant", avatar='🌿'):
         st.markdown(response)
+
+    # Convert AI response to speech
+    speech_file = text_to_speech(response)
+
+    # Play the generated speech
+    st.audio(speech_file, format="audio/mp3")
+
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
 
 # React to user input
 if prompt := textinput:
     # Display user message in chat message container
-    st.chat_message("human",avatar = "🌿").markdown(prompt)
+    st.chat_message("human", avatar="🌿").markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append({"role": "human", "content": prompt})
 
-    response = predict(message=prompt)#, temperature= temperatureSide,max_new_tokens=max_new_tokensSide, Topp=ToppSide,Repetitionpenalty=RepetitionpenaltySide)
+    response = predict(message=prompt)
+
+    # Convert AI response to speech
+    speech_file = text_to_speech(response)
+
     # Display assistant response in chat message container
     with st.chat_message("assistant", avatar='🌿'):
         st.markdown(response)
+
+    # Play the generated speech
+    st.audio(speech_file, format="audio/mp3")
+
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})