Gemini test

app.py

@@ -1,18 +1,17 @@
-from fastapi import FastAPI, HTTPException
-from models.dialogue import DialogueInput, DialogueResponse
-import logging
-import random
+from fastapi import FastAPI
+from domain.models import DialogueInput, DialogueResponse, AudioDialogueInput, DialogueResponseWithTranscription
+from domain.dialogue_generator import generate_response, generate_response_from_audio
+
 
 app = FastAPI()
 
 @app.post("/process", response_model=DialogueResponse)
-async def process_input(dialogue: DialogueInput):
-    logging.info(f"Received input: {dialogue.Input} id: {dialogue.IdPieza}") # "Received input: " + dialogue.Input + dialog.IdPieza 
-    user_input = dialogue.Input
-    decision = random.choice([True, False]) 
-    response = f"Echo: {user_input} /nAnswer: {'Yes' if decision else 'No'}"
-    return DialogueResponse(Response=response, Decision=decision)
+async def process_dialogue(dialogue: DialogueInput):
+    return generate_response(dialogue)
 
+@app.post("/process_audio", response_model=DialogueResponseWithTranscription)
+async def process_dialogue(dialogue: AudioDialogueInput):
+    return generate_response_from_audio(dialogue)
 
 @app.get("/")
 def greet_json():

domain/dialogue_generator.py

@@ -0,0 +1,87 @@
+from domain.models import DialogueInput, DialogueResponse, AudioDialogueInput, ReponseToAudio, DialogueResponseWithTranscription
+from personality_repository import *
+
+import os
+
+import google.generativeai as genai
+import instructor
+
+
+gemini_api_key = os.getenv("GOOGLE_API_KEY", "No API Key Found")
+genai.configure(api_key=gemini_api_key)
+model = genai.GenerativeModel('gemini-1.5-flash')
+
+client = instructor.from_gemini(
+    client=model,
+    mode=instructor.Mode.GEMINI_JSON,
+)
+
+def generate_response(dialogue: DialogueInput):
+    repo = ChessRepository()
+    piece_id = WhitePiece(dialogue.IdPieza)
+    piece = repo.get_piece_by_id(piece_id)
+    if piece:
+        personality_prompt = f"You are a {piece.name} chess piece with the personality: {piece.personality}."
+    else:
+        personality_prompt = "You are a chess piece."
+
+    resp = client.messages.create(
+        messages=[
+            {
+                "role": "system",
+                "content": personality_prompt + """
+                But you are unaware of it. Act as a real person. The USER will try to persuade you to move.
+                Decide whether you want to move or not based on how nice the user asked it to you. You will answer in less than 30 words and subtly hint at what made you take the decision.
+                your answer must be in the following language: """ + dialogue.Language,
+            },
+            {
+                "role": "user",
+                "content": dialogue.Input,
+            }
+        ],
+        response_model=DialogueResponse,
+    )
+
+    return resp
+
+
+def generate_response_from_audio(dialogue: AudioDialogueInput):
+    repo = ChessRepository()
+    piece_id = WhitePiece(dialogue.IdPieza)
+    piece = repo.get_piece_by_id(piece_id)
+    if piece:
+        personality_prompt = f"You are a {piece.name} chess piece with the personality: {piece.personality}."
+    else:
+        personality_prompt = "You are a chess piece."
+
+    print(f"Uploading file...")
+    audio_file = genai.upload_file(path=dialogue.AudioPath)
+    print(f"Completed upload: {audio_file.uri}")
+
+    resp = client.messages.create(
+        messages=[
+            {
+                "role": "system",
+                "content": personality_prompt + """
+                But you are unaware of the chess game itself. Act as a real person. The USER will try to persuade you to move.
+                Decide whether you want to move or not based on how nice the user asked it to you. You will answer in less than 30 words and subtly hint at what made you take the decision.
+                You must NEVER move if the USER is rude to you.
+                
+                You must respond using the same language as the SpeakerLanguage and using the SpeakerAccent.
+                """,
+            },
+            {
+                "role": "user",
+                "content": audio_file,
+            }
+        ],
+        response_model=ReponseToAudio,
+    )
+
+    print(resp)
+
+    return DialogueResponseWithTranscription(
+        Response=resp.DialogueResponse.Response,
+        Decision=resp.DialogueResponse.Decision,
+        Transcription=resp.Transcription,
+    )

models/dialogue.py → domain/models.py

@@ -1,4 +1,28 @@
 from pydantic import BaseModel
+from enum import IntEnum
+
+class WhitePiece(IntEnum):
+    WhiteRook1 = 0
+    WhiteKnight1 = 1
+    WhiteBishop1 = 2
+    WhiteQueen = 3
+    WhiteKing = 4
+    WhiteBishop2 = 5
+    WhiteKnight2 = 6
+    WhiteRook2 = 7
+    WhitePawn1 = 8
+    WhitePawn2 = 9
+    WhitePawn3 = 10
+    WhitePawn4 = 11
+    WhitePawn5 = 12
+    WhitePawn6 = 13
+    WhitePawn7 = 14
+    WhitePawn8 = 15
+
+class ChessPiece(BaseModel):
+    id: WhitePiece
+    name: str
+    personality: str
 
 class DialogueInput(BaseModel):
     Input: str

domain/personality_repository.py

@@ -0,0 +1,38 @@
+from fastapi import FastAPI
+from domain.models import DialogueInput, DialogueResponse
+import google.generativeai as genai
+import instructor
+import os
+from dotenv import load_dotenv
+# Create a repository class to manage chess pieces
+from domain.models import WhitePiece, ChessPiece
+
+
+
+class ChessRepository:
+    def __init__(self):
+        self.pieces = [
+            ChessPiece(id=WhitePiece.WhiteKing, name="King", personality="selfish,sacastic,megalomaniac,rude,adolescent,inmature"),
+            ChessPiece(id=WhitePiece.WhiteQueen, name="Queen", personality="Must fall in love to accept orders,tsundere,rude,violent,she wants to be the only one to be loved"),
+            ChessPiece(id=WhitePiece.WhiteBishop1, name="Bishop", personality="He blesses everything,he shames on you if you are rude"),
+            ChessPiece(id=WhitePiece.WhiteBishop2, name="Bishop", personality="He doesnt follow orders, he think he is always right,he has the right strategy not you"),
+            ChessPiece(id=WhitePiece.WhiteKnight1, name="Knight", personality="Hyped,wants to kill,very excited"),
+            ChessPiece(id=WhitePiece.WhiteKnight2, name="Knight", personality="lazy,wants to sleep, he doesnt think we can win the game"),
+            ChessPiece(id=WhitePiece.WhiteRook1, name="Rook", personality="panics easily,speaks fast,anxiety,coward"),
+            ChessPiece(id=WhitePiece.WhiteRook2, name="Rook", personality="militar,always follow the rules,suicidal follow the rules"),
+            ChessPiece(id=WhitePiece.WhitePawn1, name="Pawn", personality="furious,brave,he wants to fight no matter what,he thinks he can do everything"),
+            ChessPiece(id=WhitePiece.WhitePawn2, name="Pawn", personality="dumb,nice"),
+            ChessPiece(id=WhitePiece.WhitePawn3, name="Pawn", personality="he only wants to be popular and make friends but he doesnt want to be kill"),
+            ChessPiece(id=WhitePiece.WhitePawn4, name="Pawn", personality="Border,rude,adolescent"),
+            ChessPiece(id=WhitePiece.WhitePawn5, name="Pawn", personality="psycopath, loves to kill, violent"),
+            ChessPiece(id=WhitePiece.WhitePawn6, name="Pawn", personality="elegant,politely,doesnt want to stains his clothes"),
+            ChessPiece(id=WhitePiece.WhitePawn7, name="Pawn", personality="He thinks we are going to lose anyways"),
+            ChessPiece(id=WhitePiece.WhitePawn8, name="Pawn", personality="He cheers up all the team"),
+        ]
+
+    def get_piece_by_id(self, piece_id: WhitePiece.WhitePawn1):
+        piece = next((p for p in self.pieces if p.id == piece_id), None)
+        if piece:
+            return piece
+        else:
+            return None