Update app.py

app.py

@@ -25,48 +25,75 @@ print("\nEnd import library\n=========")
 print("=========\nBegin definition Backend Logic\n")
 
 print("Create default API settings")
-google_API_key=os.getenv("GEMINI_API_KEY")
+gemini_API_key=os.getenv("GEMINI_API_KEY")
 
 client = genai.Client(
-                        api_key=google_API_key,
+                        api_key=gemini_API_key,
                         http_options=genai.types.HttpOptions(api_version='v1alpha'),
                         )
 print("Set default model")
 used_model = "gemini-2.5-flash-preview-04-17"
 
-print("Define test response work")
+print("define class format_history")
+class format_history:
+    @staticmethod
+    def format_for_gemini(history):
+        print("Format history")
+        gemini_history = []
+    
+        for message_to_check in history:    
+            if message_to_check.get("role") == "user" :
+                gemini_history.append({
+                "role": "user",
+                "parts": [{"text": message_to_check.get("content", "")}]
+                })
+            elif message_to_check.get("role") == "assistant" :
+                gemini_history.append({
+                "role": "model",
+                "parts": [{"text": message_to_check.get("content", "")}]
+                })
+        return gemini_history
+
+print("define class gemini api actions")
+class gemini_API_actions:
+    apy_key = ""
+    used_model = ""
+    
+    @staticmethod
+    def set_used_model(new_model: str):
+        used_model = new_model
+        return
+
+    @staticmethod
+    def set_API_key(new_api_key: str):
+        apy_key = new_api_key
+        return
+
+    @staticmethod
+    def response_to_model(user_message, history):
+        print(f"\n=========\nUser message\n{user_message}\n")
+
+        formated_history = format_history.format_for_gemini(history)
+        
+        print(f"his: {formated_history}")
+        print("Create chat")
+        chat = client.chats.create(model=this.used_model, history=formated_history)
+        
+        print("Start response")
+        response = chat.send_message(message["text"])
+        print(f"\nResponse\n{response}=========\n")
+        
+        return response.text
+
+print("Define response work body")
 def model_response(message, history):
     print(f"\n=========\nUser message\n{message}\n")
 
-    print("Format history")
-    gemini_history = []
-
-    for message_to_check in history:
-        #print(f"t1 {message}")
-        # Skip thinking messages (messages with metadata)
-        #if not (message.get("role") == "assistant" and "metadata" in message):
-        #    print(f"t2 {message}")
-        #    formatted_history.append({
-        #        "role": "user" if message.get("role") == "user" else "assistant",
-        #        "parts": [message.get("content", "")]
-        #    })
-            
-        #print(f"t2 {message}")
-
-        if message_to_check.get("role") == "user" :
-            gemini_history.append({
-            "role": "user",
-            "parts": [{"text": message_to_check.get("content", "")}]
-            })
-        elif message_to_check.get("role") == "assistant" :
-            gemini_history.append({
-            "role": "model",
-            "parts": [{"text": message_to_check.get("content", "")}]
-            })
-
-    print(f"his: {gemini_history}")
+    formated_history = format_history.format_for_gemini(history)
+
+    print(f"his: {formated_history}")
     print("Create chat")
-    chat = client.chats.create(model=used_model, history=gemini_history)
+    chat = client.chats.create(model=used_model, history=formated_history)
 
     print("Start response")
     response = chat.send_message(message["text"])