Update app.py

app.py

@@ -25,7 +25,7 @@ model = AutoModel.from_pretrained("bert-base-uncased")
 dimension = 768  # Size of BERT embeddings
 index = faiss.IndexFlatL2(dimension)
 
-# Folder path containing agriculture-related PDFs
+# Folder path containing PDFs
 pdf_folder_path = "agriculture_pdfs"
 
 # Function to convert audio file to text
@@ -138,13 +138,12 @@ def process_audio(audio_file):
 
         if not combined_text.strip():
             return "No relevant information found in the PDFs", None
-
-        # Modify the prompt to reflect the agricultural domain
         prompt = (
-            f"The user has provided the following agriculture-related query: {combined_text}. "
+      f"The user has provided the following agriculture-related query: {combined_text}. "
             "Based on this, provide 4 clear and actionable steps to address the agricultural issue. "
             "The response should be concise and no longer than 5 lines."
-        )
+       )
+
 
         print(f"Prompt: {prompt}")  # Debugging line
 
@@ -158,11 +157,37 @@ def process_audio(audio_file):
             model="llama-3.1-70b-versatile",
         )
         response = chat_completion.choices[0].message.content
-        output_file = "agriculture_advice.mp3"
+        output_file = "advice.mp3"
         output_path = text_to_speech(response, output_file)
 
         if output_path is None:
             return "Error generating speech output", None
 
-        return response
- 
+        return response, output_path
+    except Exception as e:
+        print(f"Error in process_audio: {e}")
+        return "An error occurred while processing the audio", None
+
+iface = gr.Interface(
+    fn=process_audio,
+    inputs=gr.Audio(type="filepath"),
+    outputs=[gr.Textbox(label="Advice"), gr.Audio(label="Advice Audio")],
+    title="Kisaan Dost",
+    description="Helping Farmers about their Crops",
+    theme="default",
+    article=(
+        "<div style='text-align: center; color: #001f43;'>"
+        "<h1>Chill Mama</h1>"
+        "<h3 style='font-size: 24px;'>Helping Parents to Nurture Healthy Families</h3>"
+        "</div>"
+    ),
+    css=(
+        "<style>"
+        "body { background-color: #BAE5F4; color: #80826; }"
+        "h1, h3 { color: #001f43; }"
+        "</style>"
+    )
+)
+
+if __name__ == "__main__":
+    iface.launch()