Upload 7 files

.dockerignore

@@ -0,0 +1,5 @@
+__pycache__
+*.pyc
+.git
+.venv
+uploads/

.env

@@ -0,0 +1,2 @@
+GOOGLE_API_KEY=AIzaSyAHpElgh7EQ8SvLf7Sa2HwK-TtjOd6ZcqU
+ASSEMBLYAI_API_KEY=ya54c7e8b00f746a4bdb501a312a197e7

.gitignore

@@ -0,0 +1,3 @@
+# Ignore the virtual environment directory
+venv/
+.env

Dockerfile

@@ -0,0 +1,21 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the current directory contents into the container at /app
+COPY . /app
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Make port 7860 available to the world outside this container
+EXPOSE 7860
+
+# Define environment variables if needed
+ENV GOOGLE_API_KEY = "AIzaSyAHpElgh7EQ8SvLf7Sa2HwK-TtjOd6ZcqU"
+ENV ASSEMBLYAI_API_KEY = "ya54c7e8b00f746a4bdb501a312a197e7"
+
+# Run app.py when the container launches using Gunicorn
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]

README.md

@@ -1,11 +1,8 @@
----
-title: Basic Docker SDK Space
-emoji: 🐳
-colorFrom: purple
-colorTo: gray
-sdk: docker
-app_port: 7860
----
-
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Basic Docker SDK Space
+emoji: 🐳
+colorFrom: purple
+colorTo: gray
+sdk: docker
+app_port: 7860
+---

app.py

@@ -0,0 +1,118 @@
+import os
+import json
+from flask import Flask, jsonify, request, send_file, send_from_directory
+from langchain_core.messages import HumanMessage
+from langchain_google_genai import ChatGoogleGenerativeAI
+import assemblyai as aai
+import mimetypes
+from dotenv import load_dotenv
+
+
+mimetypes.add_type('application/javascript', '.js')
+mimetypes.add_type('text/css', '.css')
+
+load_dotenv()
+
+# Initialize the Flask app
+app = Flask(__name__)
+
+# Get API keys from environment variables
+ASSEMBLYAI_API_KEY = os.getenv("ASSEMBLYAI_API_KEY")
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+
+# Set AssemblyAI API key
+aai.settings.api_key = ASSEMBLYAI_API_KEY
+
+# Set Google API key for Gemini model
+os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
+
+# Define a directory to save uploaded audio files
+UPLOAD_FOLDER = 'uploads'
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+# Refined Instructions for Gemini
+GEMINI_INSTRUCTIONS = """
+The purpose of the category 'Reason for Outbound Call' is to identify the reason that a connected call was placed by the caller.
+
+Consider the following options and select the one that best matches the content of the call:
+1. Yes: The agent attempted to discuss buying, selling, trading, leasing, or test driving a vehicle. This includes any discussion of vehicle needs, interests, or potential sales, but this only applies if the call was live.
+2. No: The agent followed up on a previous purchase or had a general discussion. This option should be selected if the call does not include any sales-related discussion.
+3. No: The agent only confirmed, changed, or canceled an existing appointment. This includes any mention of scheduling, rescheduling, confirming, or canceling appointments.
+4. Correction: The call was not connected to the intended party (e.g., it was a voicemail or a one-sided message).
+
+Remember:
+- Option 1 should only be selected if there was a live conversation with the intended contact.
+- Option 4 should be selected if the call was not connected (e.g., it was a voicemail or no live interaction occurred).
+- Option 4 it is voice mail, make it option 4 
+"""
+
+# Home route to serve the index.html file
+@app.route('/')
+def home():
+    return send_file('web/index.html')
+
+# API route to handle file upload, transcription, and model interaction
+@app.route("/api/upload", methods=["POST"])
+def generate_api():
+    if request.method == "POST":
+        try:
+            # Check if an audio file was uploaded
+            if 'audio_file' not in request.files:
+                return jsonify({"error": "No audio file provided"}), 400
+
+            audio_file = request.files['audio_file']
+            if audio_file.filename == '':
+                return jsonify({"error": "No selected file"}), 400
+
+            # Save the uploaded file to the server
+            file_path = os.path.join(app.config['UPLOAD_FOLDER'], audio_file.filename)
+            audio_file.save(file_path)
+
+            # Transcribe the audio using AssemblyAI
+            transcriber = aai.Transcriber()
+            transcript = transcriber.transcribe(file_path)
+
+            # Send transcription and instructions to Gemini model
+            model = ChatGoogleGenerativeAI(model="gemini-pro")
+            message = HumanMessage(content=f"{GEMINI_INSTRUCTIONS}\n\nCall Transcription: {transcript.text}")
+            response = model.stream([message])
+
+            # Interpret the model's response to select the correct option
+            buffer = []
+            for chunk in response:
+                buffer.append(chunk.content)
+
+            result_text = ''.join(buffer).lower()
+
+            # Adjusted logic to determine the correct option
+            if ("option 4" in result_text or "voicemail" in result_text or 
+                "just wanted to reach out" in result_text or "thank you, bye" in result_text or 
+                "this is" in result_text and "see if" in result_text and not "response" in result_text):
+                selected_option = 4
+            elif "option 1" in result_text or "yes" in result_text or "vehicle needs" in result_text:
+                selected_option = 1
+            elif "option 2" in result_text:
+                selected_option = 2
+            elif "option 3" in result_text or "reschedule" in result_text or "confirm" in result_text or "cancel" in result_text:
+                selected_option = 3
+            else:
+                selected_option = "Could not determine the correct option."
+
+            # Return the transcription and selected option
+            return jsonify({
+                "transcription": transcript.text,
+                "selected_option": selected_option
+            }), 200
+
+        except Exception as e:
+            return jsonify({"error": str(e)})
+
+# Route to serve static files
+@app.route('/<path:path>')
+def serve_static(path):
+    return send_from_directory('web', path)
+
+# Run the Flask application
+if __name__ == '__main__':
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,6 @@
+Flask
+gunicorn
+langchain-google-genai
+langchain-core
+assemblyai
+python-dotenv