Add create_audio and create_text to samples/files.py

samples/chat.py

@@ -31,7 +31,7 @@ def test_chat(self):
             ]
         )
         response = chat.send_message("I have 2 dogs in my house.")
-        print(response.text) 
+        print(response.text)
         response = chat.send_message("How many paws are in my house?")
         print(response.text)
         # [END chat]
@@ -62,11 +62,14 @@ def test_chat_streaming_with_images(self):
         model = genai.GenerativeModel("gemini-1.5-flash")
         chat = model.start_chat()
 
-        response = chat.send_message("Hello, I'm interested in learning about musical instruments. Can I show you one?", stream=True)
+        response = chat.send_message(
+            "Hello, I'm interested in learning about musical instruments. Can I show you one?",
+            stream=True,
+        )
         for chunk in response:
             print(chunk.text)  # Yes.
             print("_" * 80)
-        
+
         organ = genai.upload_file(media / "organ.jpg")
         response = chat.send_message(
             ["What family of intruments does this instrument belong to?", organ], stream=True

samples/files.py

@@ -22,8 +22,8 @@
 
 
 class UnitTests(absltest.TestCase):
-    def test_files_create(self):
-        # [START files_create]
+    def test_files_create_text(self):
+        # [START files_create_text]
         myfile = genai.upload_file(media / "poem.txt")
         print(f"{myfile=}")
 
@@ -32,7 +32,29 @@ def test_files_create(self):
             [myfile, "\n\n", "Can you add a few more lines to this poem?"]
         )
         print(f"{result.text=}")
-        # [END files_create]
+        # [END files_create_text]
+
+    def test_files_create_image(self):
+        # [START files_create_image]
+        myfile = genai.upload_file(media / "Cajun_instruments.jpg")
+        print(f"{myfile=}")
+
+        model = genai.GenerativeModel("gemini-1.5-flash")
+        result = model.generate_content(
+            [myfile, "\n\n", "Can you tell me about the instruments in this photo?"]
+        )
+        print(f"{result.text=}")
+        # [END files_create_image]
+
+    def test_files_create_audio(self):
+        # [START files_create_audio]
+        myfile = genai.upload_file(media / "sample.mp3")
+        print(f"{myfile=}")
+
+        model = genai.GenerativeModel("gemini-1.5-flash")
+        result = model.generate_content([myfile, "Describe this audio clip"])
+        print(f"{result.text=}")
+        # [END files_create_audio]
 
     def test_files_create_video(self):
         # [START files_create_video]